Source code for mmlspark.automl.TuneHyperparameters

# Copyright (C) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See LICENSE in project root for information.

import sys
from pyspark import SQLContext
from pyspark import SparkContext

if sys.version >= '3':
    basestring = str

from mmlspark.automl._TuneHyperparameters import _TuneHyperparameters
from mmlspark.automl._TuneHyperparameters import _TuneHyperparametersModel
from import JavaParams
from import inherit_doc

[docs]@inherit_doc class TuneHyperparameters(_TuneHyperparameters): def _create_model(self, java_model): model = TuneHyperparametersModel() model._java_obj = java_model model._transfer_params_from_java() return model
[docs]@inherit_doc class TuneHyperparametersModel(_TuneHyperparametersModel):
[docs] def getBestModel(self): """ Returns the best model. """ return JavaParams._from_java(self._java_obj.getBestModel())
[docs] def getBestModelInfo(self): """ Returns the best model parameter info. """ return self._java_obj.getBestModelInfo()