package model

  1. Public
  2. All

Type Members

  1. class DecisionTreeModel extends Serializable

    :: Experimental :: Decision tree model for classification or regression.

  2. class GradientBoostedTreesModel extends TreeEnsembleModel

    :: Experimental :: Represents a gradient boosted trees model.

  3. class InformationGainStats extends Serializable

    :: DeveloperApi :: Information gain statistics for each split

  4. class Node extends Serializable with Logging

    :: DeveloperApi :: Node in a decision tree.

  5. class Predict extends Serializable

    Predicted value for a node

  6. class RandomForestModel extends TreeEnsembleModel

    :: Experimental :: Represents a random forest model.

  7. case class Split(feature: Int, threshold: Double, featureType: FeatureType, categories: List[Double]) extends Product with Serializable

    :: DeveloperApi :: Split applied to a feature