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 |
---|---|
static DecisionTreeNode |
NodeData.buildTree(Map<Integer,NodeData> nodes,
NodeData rootNodeData)
Build tree or sub-tree based on indices and nodes sorted map as a dictionary.
|
DecisionTreeNode |
DecisionTreeConditionalNode.getElseNode() |
DecisionTreeNode |
DecisionTreeConditionalNode.getMissingNode() |
DecisionTreeNode |
DecisionTreeModel.getRootNode()
Returns the root node.
|
DecisionTreeNode |
DecisionTreeConditionalNode.getThenNode() |
Modifier and Type | Method and Description |
---|---|
static String |
DecisionTreeTrainer.printTree(DecisionTreeNode node,
boolean pretty)
Represents DecisionTree as String.
|
void |
DecisionTreeConditionalNode.setElseNode(DecisionTreeNode elseNode) |
void |
DecisionTreeConditionalNode.setMissingNode(DecisionTreeNode missingNode) |
void |
DecisionTreeConditionalNode.setThenNode(DecisionTreeNode thenNode) |
Constructor and Description |
---|
DecisionTreeConditionalNode(int col,
double threshold,
DecisionTreeNode thenNode,
DecisionTreeNode elseNode,
DecisionTreeNode missingNode)
Constructs a new instance of decision tree conditional node.
|
DecisionTreeModel(DecisionTreeNode rootNode)
Creates the model.
|
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() . |
Follow @ApacheIgnite
Ignite Database and Caching Platform : ver. 2.10.0 Release Date : March 10 2021