package model
- Alphabetic
- Public
- Protected
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")