package model
- Alphabetic
- Public
- All
Type Members
-
class
DecisionTreeModel extends Serializable with Saveable
Decision tree model for classification or regression.
Decision tree model for classification or regression. This model stores the decision tree structure and parameters.
- Annotations
- @Since( "1.0.0" )
-
class
GradientBoostedTreesModel extends TreeEnsembleModel with Saveable
Represents a gradient boosted trees model.
Represents a gradient boosted trees model.
- Annotations
- @Since( "1.2.0" )
-
class
InformationGainStats extends Serializable
Information gain statistics for each split
Information gain statistics for each split
- Annotations
- @Since( "1.0.0" )
-
class
Node extends Serializable with Logging
Node in a decision tree.
Node in a decision tree.
About node indexing: Nodes are indexed from 1. Node 1 is the root; nodes 2, 3 are the left, right children. Node index 0 is not used.
- Annotations
- @Since( "1.0.0" )
-
class
Predict extends Serializable
Predicted value for a node
Predicted value for a node
- Annotations
- @Since( "1.2.0" )
-
class
RandomForestModel extends TreeEnsembleModel with Saveable
Represents a random forest model.
Represents a random forest model.
- Annotations
- @Since( "1.2.0" )
-
case class
Split(feature: Int, threshold: Double, featureType: FeatureType, categories: List[Double]) extends Product with Serializable
Split applied to a feature
Split applied to a feature
- feature
feature index
- threshold
Threshold for continuous feature. Split left if feature is less than or equal to threshold, else right.
- featureType
type of feature -- categorical or continuous
- categories
Split left if categorical feature value is in this set, else right.
- Annotations
- @Since( "1.0.0" )
Value Members
-
object
DecisionTreeModel extends Loader[DecisionTreeModel] with Logging with Serializable
- Annotations
- @Since( "1.3.0" )
-
object
GradientBoostedTreesModel extends Loader[GradientBoostedTreesModel] with Serializable
- Annotations
- @Since( "1.3.0" )
-
object
RandomForestModel extends Loader[RandomForestModel] with Serializable
- Annotations
- @Since( "1.3.0" )