Package

org.apache.spark.ml.h2o

algos

Permalink

package algos

Visibility
  1. Public
  2. All

Type Members

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

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

  3. trait H2ODeepLearningParams extends H2OAlgoParams[DeepLearningParameters]

    Permalink

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

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

    Permalink

    H2O GBM algorithm exposed via Spark ML pipelines.

  5. trait H2OGBMParams extends H2OSharedTreeParams[GBMParameters]

    Permalink

    Parameters for Spark's API exposing underlying H2O model.

Value Members

  1. object H2ODeepLearning extends MLReadable[H2ODeepLearning] with Serializable

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

    Permalink

Ungrouped