Source code for

# 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 import _TuneHyperparametersModel
from import JavaParams
from import inherit_doc

[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()