wip

Author: Aleksandr

modules/ml-ext/ml/xgboost-model-parser/pom.xml

@@ -70,6 +70,23 @@
                     <skip>false</skip>
                 </configuration>
             </plugin>
+
+            <plugin>
+                <groupId>org.antlr</groupId>
+                <artifactId>antlr4-maven-plugin</artifactId>
+                <version>${antlr.version}</version>
+                <executions>
+                    <execution>
+                        <goals>
+                            <goal>antlr4</goal>
+                        </goals>
+                        <configuration>
+                            <visitor>true</visitor>
+                            <listener>true</listener>
+                        </configuration>
+                    </execution>
+                </executions>
+            </plugin>
         </plugins>
     </build>
 

modules/ml-ext/ml/xgboost-model-parser/src/main/antlr4/org/apache/ignite/ml/xgboost/parser/XGBoostModel.g4

@@ -0,0 +1,32 @@
+grammar XGBoostModel;
+
+YES : 'yes' ;
+NO : 'no' ;
+MISSING : 'missing' ;
+EQ : '=' ;
+COMMA : ',' ;
+PLUS : '+' ;
+MINUS : '-' ;
+DOT : '.' ;
+EXP : 'E' | 'e' ;
+BOOSTER : 'booster' ;
+LBRACK : '[' ;
+RBRACK : ']' ;
+COLON : ':' ;
+LEAF : 'leaf' ;
+INT : (PLUS | MINUS)? [0-9]+ ;
+DOUBLE : INT DOT [0-9]* (EXP INT)?;
+STRING : [A-Za-z_][0-9A-Za-z_]+ ;
+NEWLINE : '\r' '\n' | '\n' | '\r' ;
+LT : '<' ;
+WS : ( ' ' | '\t' )+ -> skip ;
+
+xgValue : DOUBLE | INT ;
+xgHeader : BOOSTER LBRACK INT RBRACK COLON? ;
+xgNode : INT COLON LBRACK STRING LT xgValue RBRACK YES EQ INT COMMA NO EQ INT COMMA MISSING EQ INT ;
+xgLeaf : INT COLON LEAF EQ xgValue ;
+xgTree : xgHeader NEWLINE (
+    ((xgLeaf | xgNode) NEWLINE)+ ((xgLeaf | xgNode) EOF)?
+    | ((xgLeaf | xgNode) NEWLINE)* (xgLeaf | xgNode) EOF
+) ;
+xgModel : xgTree+ ;