org.dmg.pmml.TreeModel tm = tmec.convert(treeModel, root); tm.setModelName(String.valueOf(idCount)); Segment segment = new Segment(); if(treeModel.isGBDT()) { segment.setWeight(treeModel.getWeights().get(0).get(idCount) * treeModel.getTrees().size()); } else { segment.setWeight(treeModel.getWeights().get(0).get(idCount)); segment.setId("Segement" + String.valueOf(idCount)); idCount++; segment.setPredicate(new True()); segment.setModel(tm); list.add(segment);
Segment segment = new Segment(); segment.setWeight(1000d); segment.setId("Segement" + String.valueOf(idCount)); segment.setPredicate(new True()); segment.setModel(tmpmodel); list.add(segment); idCount++;