static private Node createLeafNode(String score, Predicate predicate){ return new LeafNode() .setScore(score) .setPredicate(predicate); }
Node result = new LeafNode() .setId(id) .setScore(depth + avgPathLength(size))
P prediction = nodepred.get(i); Node result = new LeafNode() .setId(id) .setScore(scoreEncoder.encode(prediction))
Node result = new LeafNode() .setPredicate(predicate);
org.dmg.pmml.tree.Node result = new LeafNode() .setId(id) .setScore(value)
org.dmg.pmml.tree.Node result = new LeafNode() .setId(id) .setScore(value)
@Override public Node unmarshal(ComplexNode value){ if(value.getRecordCount() != null){ return value; } // End if if(value.hasExtensions() || (value.getPartition() != null) || value.hasScoreDistributions() || (value.getEmbeddedModel() != null)){ return value; } Node node; if(value.hasNodes()){ node = new BranchNode() .setId(value.getId()) .setDefaultChild(value.getDefaultChild()); (node.getNodes()).addAll(value.getNodes()); } else { node = new LeafNode() .setId(value.getId()); } node .setScore(value.getScore()) .setPredicate(value.getPredicate()); return node; }
@Override public Node unmarshal(ComplexNode value){ if(value.getRecordCount() != null){ return value; } // End if if(value.hasExtensions() || (value.getPartition() != null) || value.hasScoreDistributions() || (value.getEmbeddedModel() != null)){ return value; } Node node; if(value.hasNodes()){ node = new BranchNode() .setId(value.getId()) .setDefaultChild(value.getDefaultChild()); (node.getNodes()).addAll(value.getNodes()); } else { node = new LeafNode() .setId(value.getId()); } node .setScore(value.getScore()) .setPredicate(value.getPredicate()); return node; }
Node result = new LeafNode() .setId(id) .setPredicate(predicate);
Node node2b = new LeafNode() .setId("2b") .setPredicate(new True());
@Test public void transform(){ Node node1a = new BranchNode(); Node node2a = new BranchNode(); Node node2b = new LeafNode(); node1a.addNodes(node2a, node2b); Node node3a = new LeafNode(); node2a.addNodes(node3a); assertTrue(node1a.getNodes() instanceof ArrayList); assertTrue(node2a.getNodes() instanceof ArrayList); TreeModel treeModel = new TreeModel(MiningFunction.CLASSIFICATION, new MiningSchema(), node1a); ArrayListTransformer transformer = new ArrayListTransformer(); transformer.applyTo(treeModel); assertTrue(node1a.getNodes() instanceof DoubletonList); assertTrue(node2a.getNodes() instanceof SingletonList); } }
Double value = prediction.getValue(i); Node result = new LeafNode() .setId(id) .setScore(value)