org.apache.ignite.ml.xgboost.parser.visitor

Class XGTreeDictionaryVisitor

java.lang.Object

org.antlr.v4.runtime.tree.AbstractParseTreeVisitor<T>

org.apache.ignite.ml.xgboost.parser.XGBoostModelBaseVisitor<Set<String>>

org.apache.ignite.ml.xgboost.parser.visitor.XGTreeDictionaryVisitor

All Implemented Interfaces:

org.antlr.v4.runtime.tree.ParseTreeVisitor<Set<String>>, XGBoostModelVisitor<Set<String>>

public class XGTreeDictionaryVisitor
extends XGBoostModelBaseVisitor<Set<String>>

Tree dictionary visitor that collects all feature names.

Constructor Summary

Constructors

Constructor and Description
XGTreeDictionaryVisitor()

Method Summary

Modifier and Type	Method and Description
Set<String>	visitXgTree(XGBoostModelParser.XgTreeContext ctx) Visit a parse tree produced by XGBoostModelParser.xgTree().

Methods inherited from class org.apache.ignite.ml.xgboost.parser.XGBoostModelBaseVisitor

visitXgHeader, visitXgLeaf, visitXgModel, visitXgNode, visitXgValue

Methods inherited from class org.antlr.v4.runtime.tree.AbstractParseTreeVisitor

aggregateResult, defaultResult, shouldVisitNextChild, visit, visitChildren, visitErrorNode, visitTerminal

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface org.antlr.v4.runtime.tree.ParseTreeVisitor

visit, visitChildren, visitErrorNode, visitTerminal

Constructor Detail

XGTreeDictionaryVisitor

public XGTreeDictionaryVisitor()

Method Detail

visitXgTree

public Set<String> visitXgTree(XGBoostModelParser.XgTreeContext ctx)

Visit a parse tree produced by XGBoostModelParser.xgTree().

The default implementation returns the result of calling AbstractParseTreeVisitor.visitChildren(org.antlr.v4.runtime.tree.RuleNode) on ctx.

Specified by:

visitXgTree in interface XGBoostModelVisitor<Set<String>>

Overrides:

visitXgTree in class XGBoostModelBaseVisitor<Set<String>>

Parameters:

ctx - the parse tree

Returns:

the visitor result