/** * Read interceptor value from parquet. * * @param g Interceptor group. */ private static double readLinRegInterceptor(SimpleGroup g) { return g.getDouble(0, 0); }
/** * Read interceptor value from parquet. * * @param g Interceptor group. */ private static double readSVMInterceptor(SimpleGroup g) { return g.getDouble(1, 0); }
/** * Read interceptor value from parquet. * * @param g Interceptor group. */ private static double readInterceptor(SimpleGroup g) { double interceptor; final SimpleGroup interceptVector = (SimpleGroup)g.getGroup(2, 0); final SimpleGroup interceptVectorVal = (SimpleGroup)interceptVector.getGroup(3, 0); final SimpleGroup interceptVectorValElement = (SimpleGroup)interceptVectorVal.getGroup(0, 0); interceptor = interceptVectorValElement.getDouble(0, 0); return interceptor; }
final SimpleGroup g = (SimpleGroup)recordReader.read(); int treeId = g.getInteger(0, 0); double treeWeight = g.getDouble(2, 0); treeWeightsByTreeID.put(treeId, treeWeight);
/** * Form the node data according data in parquet row. * * @param g The given group presenting the node data from Spark DT model. */ @NotNull private static SparkModelParser.NodeData extractNodeDataFromParquetRow(SimpleGroup g) { NodeData nodeData = new NodeData(); nodeData.id = g.getInteger(0, 0); nodeData.prediction = g.getDouble(1, 0); nodeData.leftChildId = g.getInteger(5, 0); nodeData.rightChildId = g.getInteger(6, 0); if (nodeData.leftChildId == -1 && nodeData.rightChildId == -1) { nodeData.featureIdx = -1; nodeData.threshold = -1; nodeData.isLeafNode = true; } else { final SimpleGroup splitGrp = (SimpleGroup)g.getGroup(7, 0); nodeData.featureIdx = splitGrp.getInteger(0, 0); nodeData.threshold = splitGrp.getGroup(1, 0).getGroup(0, 0).getDouble(0, 0); } return nodeData; }