models

Type Members

1. abstract class H2OAlgorithm[P <: Parameters, M <: H2OModel[M, _ <: hex.Model[_, P, _]]] extends Estimator[M] with MLWritable with H2OParams[P]

   

   Base class for H2O algorithm wrapper as a Spark transformer.

2. class H2ODeepLearning extends H2OAlgorithm[DeepLearningParameters, H2ODeepLearningModel] with H2ODeepLearningParams

   

   Creates H2ODeepLearning model If the key specified the training set is specified using setTrainKey, then frame with this key is used as the training frame, otherwise it uses the frame from the previous stage as the training frame

3. class H2ODeepLearningModel extends H2OModel[H2ODeepLearningModel, DeepLearningModel] with MLWritable

   

   Deep learning ML component.

4. trait H2ODeepLearningParams extends H2OParams[DeepLearningParameters]

   

   Parameters here can be set as normal and are duplicated to DeepLearningParameters H2O object

5. class H2OGBM extends H2OAlgorithm[GBMParameters, H2OGBMModel] with H2OGBMParams

   

6. class H2OGBMModel extends H2OModel[H2OGBMModel, GBMModel] with MLWritable

   

   H2O GBM Algo exposed via Spark ML pipelines.

7. trait H2OGBMParams extends H2OTreeParams[GBMParameters]

   

   Parameters for Spark's API exposing underlying H2O model.

8. abstract class H2OModel[S <: H2OModel[S, M], M <: hex.Model[_, _, _ <: Output]] extends Model[S] with MLWritable

   

   Shared implementation for H2O model pipelines

9. trait H2OParams[P <: Parameters] extends Params

   

   A trait extracting a shared parameters among all models.

10. trait H2OTreeParams[P <: SharedTreeParameters] extends H2OParams[P]