protected void initChildModels() { childModels = new HashMap<>(); Iterator<Serializable> extenIter = rawModel.getExtensionsAndMiningSchemasAndOutputs().iterator(); segmentation = null; while (extenIter.hasNext() && segmentation == null) { Object obj = extenIter.next(); if (obj instanceof Segmentation) { segmentation = new MiningSegmentation(this,(Segmentation)obj); } } if (segmentation != null) { List<MiningSegment> segments = segmentation.getMiningSegments(); DataDictionary dd = this.getDataDictionary(); for (MiningSegment seg : segmentation.getMiningSegments()) { childModels.put(seg.getModel().getModelId(), seg.getModel()); } } }
public String generateRules() { StringBuilder bldr = new StringBuilder(); if (this.scoreable && this.segmentation != null) { bldr.append(segmentation.generateSegmentationRules()); } return bldr.toString(); }
public String getSegmentationAgendaId() { return getOwner().getModelId()+"_"+getSegmentationId(); } }
public MiningSegmentation(Miningmodel owner, Segmentation segmentation) { this.owner = owner; this.multipleModelMethod = segmentation.getMultipleModelMethod(); this.miningSegments = new ArrayList<>(); initSegments(segmentation.getSegments()); initTemplates(); }
public String generateSegmentationRules() { StringBuilder builder = new StringBuilder(); loadTemplates(this.multipleModelMethod); Map<String, Object> templateVars = new HashMap<>(); String pkgName = this.getOwner().getModelPackageName(); CompiledTemplate ct = null; ByteArrayOutputStream baos = new ByteArrayOutputStream(); break; case MODEL_CHAIN: templateVars.put("miningModel", this.getOwner()); templateVars.put("childSegments", this.getMiningSegments()); templateVars.put("packageName", pkgName); templateVars.put("ruleUnitClassName", this.getOwner().getRuleUnitClassName()); ct = templates.getNamedTemplate(this.multipleModelMethod.name()); TemplateRuntime.execute(ct,null,new MapVariableResolverFactory(templateVars),baos); break; case SELECT_ALL: templateVars.put("ruleUnitClassName", this.getOwner().getRuleUnitClassName()); templateVars.put("miningModel", this.getOwner()); templateVars.put("childSegments", this.getMiningSegments()); templateVars.put("packageName", pkgName); ct = templates.getNamedTemplate(this.multipleModelMethod.name()); break; case SELECT_FIRST: templateVars.put("ruleUnitClassName", this.getOwner().getRuleUnitClassName()); templateVars.put("miningModel", this.getOwner()); templateVars.put("childSegments", this.getMiningSegments());
@Override public List<PMMLMiningField> getMiningFields() { List<PMMLMiningField> fields = super.getMiningFields(); for (PMMLMiningField field : fields) { String type = field.getType(); if (type == null || type.trim().isEmpty()) { fields.remove(field); } } if (this.segmentation != null) { segmentation.getMiningSegments().stream() .flatMap(ms -> ms.getModel().getMiningFields().stream() ) .filter(fld -> !fields.contains(fld)) .forEach(fields::add); } return fields; }
public String getSegmentId() { if (this.segmentId == null || this.segmentId.trim().isEmpty()) { StringBuilder bldr = new StringBuilder(owner.getSegmentationId()); bldr.append("Segment").append(this.segmentIndex); this.segmentId = bldr.toString(); } return this.segmentId; }
public PMML4Model getModel(Segment segment, MiningSegmentation segmentation) { PMML4Model model = null; if (segment.getMiningModel() != null) { MiningModel mm = segment.getMiningModel(); model = new Miningmodel(mm.getModelName(), mm, segmentation.getOwner(), null); } else if (segment.getRegressionModel() != null) { RegressionModel rm = segment.getRegressionModel(); model = new Regression(rm.getModelName(), rm, segmentation.getOwner(), null); } else if (segment.getScorecard() != null) { Scorecard sc = segment.getScorecard(); model = new ScorecardModel(sc.getModelName(), sc, segmentation.getOwner(), null); } else if (segment.getTreeModel() != null) { TreeModel tm = segment.getTreeModel(); model = new Treemodel(tm.getModelName(), tm, segmentation.getOwner(), null); } return model; }
protected List<PMMLResource> getChildResources(PMML pmml_origin, Miningmodel parent, List<PMMLResource> resourceList, ClassLoader classLoader, KieModuleModel module) { if (parent != null && parent.getSegmentation() != null) { MiningSegmentation segmentation = parent.getSegmentation(); if (segmentation.getMiningSegments() != null) { List<MiningSegment> segments = segmentation.getMiningSegments(); for (MiningSegment segment : segments) { if (segment.getModel() instanceof Miningmodel) { buildResourcesFromModel(pmml_origin, (Miningmodel) segment.getModel(), resourceList, classLoader, module); } else { resourceList.add(buildResourceFromSegment(pmml_origin, segment, classLoader, module)); } } } } return resourceList; }
public MiningSegmentWeight(MiningSegment segment) { super(); this.segmentationId = segment.getOwner().getSegmentationId(); this.segmentId = segment.getSegmentId(); List<String> targets = segment.getTargetsForWeighting(); if (targets != null && !targets.isEmpty()) { this.targetName = targets.get(0); } this.weight = segment.getWeight(); } public MiningSegmentWeight(String segmentationId, String segmentId, String targetName, Double weight) {
protected PMMLResource buildResourceFromSegment(PMML pmml_origin, MiningSegment segment, ClassLoader classLoader, KieModuleModel module) { PMML pmml = new PMML(); DataDictionary dd = pmml_origin.getDataDictionary(); pmml.setDataDictionary(dd); pmml.setHeader(pmml_origin.getHeader()); pmml.getAssociationModelsAndBaselineModelsAndClusteringModels().add(segment.getModel().getRawModel()); addMissingFieldDefinition(pmml, segment.getOwner(), segment); helper.setPack(segment.getModel().getModelPackageName());//PMML4Helper.pmmlDefaultPackageName()+".mining.segment_"+segment.getSegmentId()); StringBuilder rules = new StringBuilder(this.compile(pmml, classLoader)); String extBeanMiningRules = segment.getModel().getExternalBeansMiningRules(); if (extBeanMiningRules != null) { rules.append(extBeanMiningRules); } KieBaseModel kbModel = module.newKieBaseModel(segment.getOwner().getOwner().getModelId() + "_" + segment.getOwner().getSegmentationId() + "_SEGMENT_" + segment.getSegmentId()); kbModel.addPackage(helper.getPack()) .setDefault(false) .setEventProcessingMode(EventProcessingOption.CLOUD); KieSessionModel ksm = kbModel.newKieSessionModel("SEGMENT_" + segment.getSegmentId()); ksm.setDefault(true); PMMLResource resource = new PMMLResource(helper.getPack()); resource.setKieBaseModel(kbModel); resource.addRules(segment.getModel().getModelId(), rules.toString()); return resource; }
private void addMissingFieldDefinition(PMML pmml, MiningSegmentation msm, MiningSegment seg) { List<PMML4Model> models = msm.getMiningSegments().stream() .filter(s -> s != seg && s.getSegmentIndex() < seg.getSegmentIndex()) .map(iseg -> {