/** * @param model {@link KMeansModel} to translate to PMML * @return PMML representation of a KMeans cluster model */ private PMML kMeansModelToPMML(KMeansModel model, Map<Integer,Long> clusterSizesMap) { ClusteringModel clusteringModel = pmmlClusteringModel(model, clusterSizesMap); PMML pmml = PMMLUtils.buildSkeletonPMML(); pmml.setDataDictionary(AppPMMLUtils.buildDataDictionary(inputSchema, null)); pmml.addModels(clusteringModel); return pmml; }
public static PMML buildDummyModel() { Node node = new Node().setRecordCount(123.0); TreeModel treeModel = new TreeModel(MiningFunction.CLASSIFICATION, null, node); PMML pmml = PMMLUtils.buildSkeletonPMML(); pmml.addModels(treeModel); return pmml; }
@Test public void testReadPMMLFromMessage() throws Exception { PMML pmml = PMMLUtils.buildSkeletonPMML(); String pmmlString = PMMLUtils.toString(pmml); assertEquals(PMMLUtils.VERSION, AppPMMLUtils.readPMMLFromUpdateKeyMessage( "MODEL", pmmlString, null).getVersion()); Path pmmlPath = getTempDir().resolve("out.pmml"); Files.write(pmmlPath, Collections.singleton(pmmlString)); assertEquals(PMMLUtils.VERSION, AppPMMLUtils.readPMMLFromUpdateKeyMessage( "MODEL-REF", pmmlPath.toAbsolutePath().toString(), null).getVersion()); assertNull(AppPMMLUtils.readPMMLFromUpdateKeyMessage("MODEL-REF", "no-such-path", null)); }
private static PMML buildDummyModel() { Node node = new Node().setRecordCount(123.0); TreeModel treeModel = new TreeModel(MiningFunction.CLASSIFICATION, null, node); PMML pmml = PMMLUtils.buildSkeletonPMML(); pmml.addModels(treeModel); return pmml; }
@Override public Pair<String,String> generate(int id, RandomGenerator random) { if (id % 10 == 0) { PMML pmml = PMMLUtils.buildSkeletonPMML(); AppPMMLUtils.addExtension(pmml, "features", 2); AppPMMLUtils.addExtension(pmml, "implicit", true); AppPMMLUtils.addExtensionContent(pmml, "XIDs", X.keySet()); AppPMMLUtils.addExtensionContent(pmml, "YIDs", Y.keySet()); return new Pair<>("MODEL", PMMLUtils.toString(pmml)); } else { int xOrYID = id % 10; String xOrYIDString = ALSUtilsTest.idToStringID(id); String message; boolean isX = xOrYID >= 6; if (isX) { message = TextUtils.joinJSON(Arrays.asList( "X", xOrYIDString, X.get(xOrYIDString), A.get(xOrYIDString))); } else { message = TextUtils.joinJSON(Arrays.asList( "Y", xOrYIDString, Y.get(xOrYIDString), At.get(xOrYIDString))); } return new Pair<>("UP", message); } }
AppPMMLUtils.buildDataDictionary(inputSchema, categoricalValueEncodings); PMML pmml = PMMLUtils.buildSkeletonPMML(); pmml.setDataDictionary(dictionary); pmml.addModels(model);
@Test public void testSkeleton() { PMML pmml = PMMLUtils.buildSkeletonPMML(); assertEquals("Oryx", pmml.getHeader().getApplication().getName()); assertNotNull(pmml.getHeader().getTimestamp()); }
saveFeaturesRDD(itemFeaturesRDD, new Path(candidatePath, "Y"), bItemIndexToID); PMML pmml = PMMLUtils.buildSkeletonPMML(); AppPMMLUtils.addExtension(pmml, "X", "X/"); AppPMMLUtils.addExtension(pmml, "Y", "Y/");
public static PMML buildDummyClusteringModel() { PMML pmml = PMMLUtils.buildSkeletonPMML();
public static PMML buildDummyRegressionModel() { PMML pmml = PMMLUtils.buildSkeletonPMML();
private static PMML buildDummyClassificationModel(int numTrees) { PMML pmml = PMMLUtils.buildSkeletonPMML();