Package

org.apache.spark.ml.h2o

algos

Permalink

package algos

Visibility
  1. Public
  2. All

Type Members

  1. class GBMParametersParam extends Param[GBMParameters]

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

    Permalink

    Base class for H2O algorithm wrapper as a Spark transformer.

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

    Permalink

    H2O AutoML pipeline step

  4. class H2OAutoMLAlgosParam extends Param[Array[algo]]

    Permalink
  5. trait H2OAutoMLParams extends Params

    Permalink
  6. class H2OAutoMLStoppingMetricParam extends Param[StoppingMetric]

    Permalink
  7. class H2ODeepLearning extends H2OAlgorithm[DeepLearningParameters, H2OMOJOModel] with H2ODeepLearningParams

    Permalink

    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

  8. trait H2ODeepLearningParams extends H2OAlgoParams[DeepLearningParameters]

    Permalink

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

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

    Permalink

    H2O GBM algorithm exposed via Spark ML pipelines.

  10. trait H2OGBMParams extends H2OSharedTreeParams[GBMParameters]

    Permalink

    Parameters for Spark's API exposing underlying H2O model.

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

    Permalink

    H2O Grid Search, currently available just for GBM

  12. trait H2OGridSearchParams extends Params

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

    Permalink
  14. class NullableFloatArrayParam extends Param[Array[Float]]

    Permalink

Value Members

  1. object H2OAlgorithmReader

    Permalink
  2. object H2OAutoML extends MLReadable[H2OAutoML] with Serializable

    Permalink
  3. object H2OAutoMLReader

    Permalink
  4. object H2ODeepLearning extends MLReadable[H2ODeepLearning] with Serializable

    Permalink
  5. object H2OGBM extends MLReadable[H2OGBM] with Serializable

    Permalink
  6. object H2OGridSearch extends MLReadable[H2OGridSearch] with Serializable

    Permalink

Ungrouped