algos

Type Members

1. class BackendParam extends EnumParam[Backend]

   

2. class BoosterParam extends EnumParam[Booster]

   

3. class DMatrixTypeParam extends EnumParam[DMatrixType]

   

4. class DartNormalizeTypeParam extends EnumParam[DartNormalizeType]

   

5. class DartSampleTypeParam extends EnumParam[DartSampleType]

   

6. class GBMParametersParam extends Param[GBMParameters]

   

7. class GrowPolicyParam extends EnumParam[GrowPolicy]

   

8. 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.

9. class H2OAutoML extends Estimator[H2OMOJOModel] with MLWritable with H2OAutoMLParams

   

   H2O AutoML pipeline step

10. class H2OAutoMLAlgosParam extends EnumArrayParam[Algo]

    

11. trait H2OAutoMLParams extends Params

    

12. class H2OAutoMLStoppingMetricParam extends EnumParam[StoppingMetric]

    

13. 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

14. trait H2ODeepLearningParams extends H2OAlgoParams[DeepLearningParameters]

    

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

15. class H2OGBM extends H2OAlgorithm[GBMParameters, H2OMOJOModel] with H2OGBMParams

    

    H2O GBM algorithm exposed via Spark ML pipelines.

16. trait H2OGBMParams extends H2OSharedTreeParams[GBMParameters]

    

    Parameters for Spark's API exposing underlying H2O model.

17. class H2OGridSearch extends Estimator[H2OMOJOModel] with MLWritable with H2OGridSearchParams

    

    H2O Grid Search, currently available just for GBM

18. trait H2OGridSearchParams extends Params

    

19. class H2OXGBoost extends H2OAlgorithm[XGBoostParameters, H2OMOJOModel] with H2OXGBoostParams

    

    H2O XGBoost algorithm exposed via Spark ML pipelines.

20. trait H2OXGBoostParams extends H2OAlgoParams[XGBoostParameters]

    

    Parameters for Spark's API exposing underlying H2O model.

21. class HyperParamsParam extends Param[Map[String, Array[AnyRef]]]

    

22. class MissingValuesHandlingParam extends EnumParam[MissingValuesHandling]

    

23. class TreeMethodParam extends EnumParam[TreeMethod]