algos

Type Members

class BackendParam extends EnumParam[Backend]
class BoosterParam extends EnumParam[Booster]
class DMatrixTypeParam extends EnumParam[DMatrixType]
class DartNormalizeTypeParam extends EnumParam[DartNormalizeType]
class DartSampleTypeParam extends EnumParam[DartSampleType]
class GrowPolicyParam extends EnumParam[GrowPolicy]
abstract class H2OAlgorithm[P <: Parameters, M <: Model[M]] extends Estimator[M] with MLWritable with H2OAlgoParams[P]

Base class for H2O algorithm wrapper as a Spark transformer.
class H2OAutoML extends Estimator[H2OMOJOModel] with MLWritable with H2OAutoMLParams

H2O AutoML pipeline step
class H2OAutoMLAlgosParam extends EnumArrayParam[Algo]
trait H2OAutoMLParams extends Params
class H2OAutoMLStoppingMetricParam extends EnumParam[StoppingMetric]
class H2ODeepLearning extends H2OAlgorithm[DeepLearningParameters, H2OMOJOModel] with H2ODeepLearningParams

H2O Deep learning algorithm exposed via Spark ML pipelines.
H2O Deep learning algorithm exposed via Spark ML pipelines.
TODO: There are still bunch of parameters defined DeepLearningParameters which need to be ported here
trait H2ODeepLearningParams extends H2OAlgoParams[DeepLearningParameters]

Parameters here can be set as normal and are duplicated to DeepLearningParameters H2O object
class H2OGBM extends H2OAlgorithm[GBMParameters, H2OMOJOModel] with H2OGBMParams

H2O GBM algorithm exposed via Spark ML pipelines.
trait H2OGBMParams extends H2OSharedTreeParams[GBMParameters]

Parameters for Spark's API exposing underlying H2O model.
class H2OGLM extends H2OAlgorithm[GLMParameters, H2OMOJOModel] with H2OGLMParams

H2O GLM algorithm exposed via Spark ML pipelines.
class H2OGLMFamilyParam extends EnumParam[Family]
class H2OGLMLinkParam extends EnumParam[Link]
class H2OGLMMissingValuesHandlingParam extends EnumParam[MissingValuesHandling]
trait H2OGLMParams extends H2OAlgoParams[GLMParameters]

Parameters for Spark's API exposing underlying H2O model.
class H2OGLMSolverParam extends EnumParam[Solver]
class H2OGLMStringPairArrayParam extends Param[Array[(String, String)]]
class H2OGridSearch extends Estimator[H2OMOJOModel] with MLWritable with H2OGridSearchParams

H2O Grid Search, currently available just for GBM
trait H2OGridSearchParams extends Params
class H2OXGBoost extends H2OAlgorithm[XGBoostParameters, H2OMOJOModel] with H2OXGBoostParams

H2O XGBoost algorithm exposed via Spark ML pipelines.
trait H2OXGBoostParams extends H2OAlgoParams[XGBoostParameters]

Parameters for Spark's API exposing underlying H2O model.
class MissingValuesHandlingParam extends EnumParam[MissingValuesHandling]
class TreeMethodParam extends EnumParam[TreeMethod]

Value Members

object H2OAlgorithmReader
object H2OAutoML extends MLReadable[H2OAutoML] with Serializable
object H2OAutoMLReader
object H2ODeepLearning extends MLReadable[H2ODeepLearning] with Serializable
object H2OGBM extends MLReadable[H2OGBM] with Serializable
object H2OGLM extends MLReadable[H2OGLM] with Serializable
object H2OGridSearch extends MLReadable[H2OGridSearch] with Serializable
object H2OGridSearchReader
object H2OXGBoost extends MLReadable[H2OXGBoost] with Serializable

package algos

Type Members

class BackendParam extends EnumParam[Backend]

class BoosterParam extends EnumParam[Booster]

class DMatrixTypeParam extends EnumParam[DMatrixType]

class DartNormalizeTypeParam extends EnumParam[DartNormalizeType]

class DartSampleTypeParam extends EnumParam[DartSampleType]

class GrowPolicyParam extends EnumParam[GrowPolicy]

abstract class H2OAlgorithm[P <: Parameters, M <: Model[M]] extends Estimator[M] with MLWritable with H2OAlgoParams[P]

class H2OAutoML extends Estimator[H2OMOJOModel] with MLWritable with H2OAutoMLParams

class H2OAutoMLAlgosParam extends EnumArrayParam[Algo]

trait H2OAutoMLParams extends Params

class H2OAutoMLStoppingMetricParam extends EnumParam[StoppingMetric]

class H2ODeepLearning extends H2OAlgorithm[DeepLearningParameters, H2OMOJOModel] with H2ODeepLearningParams

trait H2ODeepLearningParams extends H2OAlgoParams[DeepLearningParameters]

class H2OGBM extends H2OAlgorithm[GBMParameters, H2OMOJOModel] with H2OGBMParams

trait H2OGBMParams extends H2OSharedTreeParams[GBMParameters]

class H2OGLM extends H2OAlgorithm[GLMParameters, H2OMOJOModel] with H2OGLMParams

class H2OGLMFamilyParam extends EnumParam[Family]

class H2OGLMLinkParam extends EnumParam[Link]

class H2OGLMMissingValuesHandlingParam extends EnumParam[MissingValuesHandling]

trait H2OGLMParams extends H2OAlgoParams[GLMParameters]

class H2OGLMSolverParam extends EnumParam[Solver]

class H2OGLMStringPairArrayParam extends Param[Array[(String, String)]]

class H2OGridSearch extends Estimator[H2OMOJOModel] with MLWritable with H2OGridSearchParams

trait H2OGridSearchParams extends Params

class H2OXGBoost extends H2OAlgorithm[XGBoostParameters, H2OMOJOModel] with H2OXGBoostParams

trait H2OXGBoostParams extends H2OAlgoParams[XGBoostParameters]

class MissingValuesHandlingParam extends EnumParam[MissingValuesHandling]

class TreeMethodParam extends EnumParam[TreeMethod]

Value Members

object H2OAlgorithmReader

object H2OAutoML extends MLReadable[H2OAutoML] with Serializable

object H2OAutoMLReader

object H2ODeepLearning extends MLReadable[H2ODeepLearning] with Serializable

object H2OGBM extends MLReadable[H2OGBM] with Serializable

object H2OGLM extends MLReadable[H2OGLM] with Serializable

object H2OGridSearch extends MLReadable[H2OGridSearch] with Serializable

object H2OGridSearchReader

object H2OXGBoost extends MLReadable[H2OXGBoost] with Serializable

Ungrouped