Package | Description |
---|---|
org.apache.ignite.ml.tree |
Root package for decision trees.
|
org.apache.ignite.ml.xgboost |
Base package for XGBoost model parser, correspondent DTOs and util classes.
|
org.apache.ignite.ml.xgboost.parser.visitor |
Base package for XGBoost model parser ANTLR visitor.
|
Modifier and Type | Class and Description |
---|---|
class |
DecisionTreeConditionalNode
Decision tree conditional (non-leaf) node.
|
class |
DecisionTreeLeafNode
Decision tree leaf node which contains value.
|
Modifier and Type | Method and Description |
---|---|
<K,V> DecisionTreeNode |
DecisionTree.fit(Dataset<EmptyContext,DecisionTreeData> dataset) |
<K,V> DecisionTreeNode |
DecisionTree.fitWithInitializedDeployingContext(DatasetBuilder<K,V> datasetBuilder,
Preprocessor<K,V> preprocessor)
Trains model based on the specified data.
|
DecisionTreeNode |
DecisionTreeConditionalNode.getElseNode() |
DecisionTreeNode |
DecisionTreeConditionalNode.getMissingNode() |
DecisionTreeNode |
DecisionTreeConditionalNode.getThenNode() |
protected <K,V> DecisionTreeNode |
DecisionTree.updateModel(DecisionTreeNode mdl,
DatasetBuilder<K,V> datasetBuilder,
Preprocessor<K,V> preprocessor)
Trains new model based on dataset because there is no valid approach to update decision trees.
|
Modifier and Type | Method and Description |
---|---|
boolean |
DecisionTree.isUpdateable(DecisionTreeNode mdl) |
static String |
DecisionTree.printTree(DecisionTreeNode node,
boolean pretty)
Represents DecisionTree as String.
|
void |
DecisionTreeConditionalNode.setElseNode(DecisionTreeNode elseNode) |
void |
DecisionTreeConditionalNode.setMissingNode(DecisionTreeNode missingNode) |
void |
DecisionTreeConditionalNode.setThenNode(DecisionTreeNode thenNode) |
protected <K,V> DecisionTreeNode |
DecisionTree.updateModel(DecisionTreeNode mdl,
DatasetBuilder<K,V> datasetBuilder,
Preprocessor<K,V> preprocessor)
Trains new model based on dataset because there is no valid approach to update decision trees.
|
Constructor and Description |
---|
DecisionTreeConditionalNode(int col,
double threshold,
DecisionTreeNode thenNode,
DecisionTreeNode elseNode,
DecisionTreeNode missingNode)
Constructs a new instance of decision tree conditional node.
|
Constructor and Description |
---|
XGModelComposition(Map<String,Integer> dict,
List<DecisionTreeNode> models)
Constructs a new instance of composition of models.
|
Modifier and Type | Method and Description |
---|---|
DecisionTreeNode |
XGTreeVisitor.visitXgTree(XGBoostModelParser.XgTreeContext ctx)
Visit a parse tree produced by
XGBoostModelParser.xgTree() . |
GridGain In-Memory Computing Platform : ver. 8.9.15 Release Date : December 3 2024