Package

ai.h2o.sparkling.ml

algos

Permalink

package algos

Visibility
  1. Public
  2. All

Type Members

  1. trait H2OAlgoCommonUtils extends H2OCommonParams

    Permalink
  2. abstract class H2OAlgorithm[B <: H2OBaseModelBuilder, M <: H2OBaseModel, P <: Parameters] extends Estimator[H2OMOJOModel] with H2OAlgoCommonUtils with DefaultParamsWritable with H2OAlgoCommonParams[P]

    Permalink

    Base class for H2O algorithm wrapper as a Spark transformer.

  3. class H2OAutoML extends Estimator[H2OMOJOModel] with H2OAlgoCommonUtils with DefaultParamsWritable with H2OAutoMLParams

    Permalink

    H2O AutoML pipeline step

  4. trait H2OAutoMLParams extends H2OCommonSupervisedParams

    Permalink
  5. final class H2OAutoMLSortMetric extends Enum[H2OAutoMLSortMetric]

    Permalink
  6. class H2ODeepLearning extends H2OSupervisedAlgorithm[DeepLearning, DeepLearningModel, DeepLearningParameters] 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

  7. trait H2ODeepLearningParams extends H2OAlgoSupervisedParams[DeepLearningParameters]

    Permalink

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

  8. class H2OGBM extends H2OSupervisedAlgorithm[GBM, GBMModel, GBMParameters] with H2OGBMParams

    Permalink

    H2O GBM algorithm exposed via Spark ML pipelines.

  9. trait H2OGBMParams extends H2OAlgoSharedTreeParams[GBMParameters]

    Permalink

    Parameters for Spark's API exposing underlying H2O model.

  10. class H2OGLM extends H2OSupervisedAlgorithm[GLM, GLMModel, GLMParameters] with H2OGLMParams

    Permalink

    H2O GLM algorithm exposed via Spark ML pipelines.

  11. trait H2OGLMParams extends H2OAlgoSupervisedParams[GLMParameters]

    Permalink

    Parameters for Spark's API exposing underlying H2O model.

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

    Permalink
  13. class H2OGridSearch extends Estimator[H2OMOJOModel] with H2OAlgoCommonUtils with DefaultParamsWritable with H2OGridSearchParams

    Permalink

    H2O Grid Search

  14. final class H2OGridSearchMetric extends Enum[H2OGridSearchMetric]

    Permalink
  15. trait H2OGridSearchParams extends H2OCommonSupervisedParams

    Permalink
  16. class H2OKMeans extends H2OUnsupervisedAlgorithm[KMeans, KMeansModel, KMeansParameters] with H2OKMeansParams

    Permalink

    H2O KMeans algorithm exposed via Spark ML pipelines.

  17. trait H2OKMeansParams extends H2OAlgoUnsupervisedParams[KMeansParameters]

    Permalink

    Parameters for Spark's API exposing underlying H2O model.

  18. abstract class H2OSupervisedAlgorithm[B <: H2OBaseModelBuilder, M <: H2OBaseModel, P <: Parameters] extends H2OAlgorithm[B, M, P] with H2OAlgoSupervisedParams[P]

    Permalink
  19. abstract class H2OUnsupervisedAlgorithm[B <: H2OBaseModelBuilder, M <: H2OBaseModel, P <: Parameters] extends H2OAlgorithm[B, M, P] with H2OAlgoUnsupervisedParams[P]

    Permalink
  20. class H2OXGBoost extends H2OSupervisedAlgorithm[XGBoost, XGBoostModel, XGBoostParameters] with H2OXGBoostParams

    Permalink

    H2O XGBoost algorithm exposed via Spark ML pipelines.

  21. trait H2OXGBoostParams extends H2OAlgoSupervisedParams[XGBoostParameters] with DeprecatableParams

    Permalink

    Parameters for Spark's API exposing underlying H2O model.

Value Members

  1. object H2OAutoML extends H2OParamsReadable[H2OAutoML] with Serializable

    Permalink
  2. object H2ODeepLearning extends H2OParamsReadable[H2ODeepLearning] with Serializable

    Permalink
  3. object H2OGBM extends H2OParamsReadable[H2OGBM] with Serializable

    Permalink
  4. object H2OGLM extends H2OParamsReadable[H2OGLM] with Serializable

    Permalink
  5. object H2OGridSearch extends H2OParamsReadable[H2OGridSearch] with Serializable

    Permalink
  6. object H2OKMeans extends DefaultParamsReadable[H2OKMeans] with Serializable

    Permalink
  7. object H2OXGBoost extends H2OParamsReadable[H2OXGBoost] with Serializable

    Permalink

Ungrouped