Python ComputeModelStatistics.limit Examples

Programming Language: Python

Namespace/Package Name: mmlspark.train

Method/Function: limit

Examples at hotexamples.com: 1

Python ComputeModelStatistics.limit - 1 examples found. These are the top rated real world Python examples of mmlspark.train.ComputeModelStatistics.limit extracted from open source projects. You can rate examples to help us improve the quality of examples.

Frequently Used Methods

Show Hide

ComputeModelStatistics(9)

toPandas(3)

first(2)

limit(2)

show(1)

Example #1

Show file

models = [lg, rf, gbt]
mml_models = [
    TrainClassifier(model=model, labelCol="label") for model in models
]
param_builder = HyperparamBuilder() \
    .addHyperparam(lg, lg.regParam, RangeHyperParam(0.1, 0.3)) \
    .addHyperparam(rf, rf.numTrees, DiscreteHyperParam([5, 10])) \
    .addHyperparam(rf, rf.maxDepth, DiscreteHyperParam([3, 5])) \
    .addHyperparam(gbt, gbt.maxBins, RangeHyperParam(8, 16)) \
    .addHyperparam(gbt, gbt.maxDepth, DiscreteHyperParam([3, 5]))

search_space = param_builder.build()
print(search_space)
random_space = RandomSpace(search_space)

best_model = TuneHyperparameters(evaluationMetric="accuracy",
                                 models=mml_models,
                                 numFolds=2,
                                 numRuns=len(mml_models) * 2,
                                 parallelism=1,
                                 paramSpace=random_space.space(),
                                 seed=0).fit(train_data)

print(best_model.getBestModelInfo())
print(best_model.getBestModel())

prediction = best_model.transform(test_data)
metrics = ComputeModelStatistics().transform(prediction)
print(metrics.limit(10).toPandas())