public String getSegmentPackageName() { StringBuilder builder = new StringBuilder(segmentPackageRootName); builder.append(".mining.segment_").append(this.getSegmentId()); return builder.toString(); }
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) {
public MiningSegmentTransfer getSegmentTransfer( MiningSegment targetSegment, String targetFieldName) { MiningSegmentTransfer xfer = null; int lastIndex = targetSegment.getSegmentIndex() - 1; if (lastIndex < 0) { throw new IndexOutOfBoundsException("Cannot have an undefined mining field in the first segment"); } for (int idx = lastIndex; idx >= 0; idx--) { MiningSegment seg = miningSegments.get(idx); if (seg != null && seg.getInternalModel() != null) { List<PMMLOutputField> outputs = seg.getInternalModel().getOutputFields(); if (outputs != null) { for (PMMLOutputField field: outputs) { if (field.getName().equals(targetFieldName)) { xfer = new MiningSegmentTransfer(this.segmentationId, seg.getSegmentId(), targetSegment.getSegmentId()); xfer.addResultToRequestMapping(targetFieldName, targetFieldName); return xfer; } } } } } return xfer; }
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; }