public Output addOutputFields(OutputField... outputFields) { getOutputFields().addAll(Arrays.asList(outputFields)); return this; }
public Output addOutputFields(OutputField... outputFields) { getOutputFields().addAll(Arrays.asList(outputFields)); return this; }
@Override public Map<FieldName, org.dmg.pmml.OutputField> load(Output output){ return IndexableUtil.buildMap(output.getOutputFields()); } });
@Override public VisitorAction visit(Output output){ if(output.hasOutputFields()){ this.outputFields.addAll(output.getOutputFields()); } return super.visit(output); }
@Override public VisitorAction visit(Output output){ if(output.hasOutputFields()){ this.outputFields.addAll(output.getOutputFields()); } return super.visit(output); }
@Override public VisitorAction visit(Output output){ if(output.hasOutputFields()){ declare(output, output.getOutputFields()); suppress(output.getOutputFields()); } return super.visit(output); }
@Override public VisitorAction visit(Output output){ if(output.hasOutputFields()){ declare(output, output.getOutputFields()); suppress(output.getOutputFields()); } return super.visit(output); }
protected List<OutputField> createOutputFields(){ M model = getModel(); Output output = model.getOutput(); List<OutputField> resultFields = new ArrayList<>(); if(output != null && output.hasOutputFields()){ List<org.dmg.pmml.OutputField> outputFields = output.getOutputFields(); Predicate<org.dmg.pmml.OutputField> outputFilter = ensureOutputFilter(); outputFields: for(org.dmg.pmml.OutputField outputField : outputFields){ if(outputFilter.test(outputField)){ OutputField resultField = new OutputField(outputField); resultFields.add(resultField); } } } return ImmutableList.copyOf(resultFields); }
protected Set<org.dmg.pmml.ResultFeature> collectResultFeatures(){ M model = getModel(); Output output = model.getOutput(); Set<org.dmg.pmml.ResultFeature> resultFeatures = EnumSet.noneOf(org.dmg.pmml.ResultFeature.class); if(output != null && output.hasOutputFields()){ List<org.dmg.pmml.OutputField> outputFields = output.getOutputFields(); for(org.dmg.pmml.OutputField outputField : outputFields){ resultFeatures.add(outputField.getResultFeature()); } } return Sets.immutableEnumSet(resultFeatures); }
static public OutputField getOutputField(Output output, FieldName name){ if(output.hasOutputFields()){ List<OutputField> outputFields = output.getOutputFields(); for(OutputField outputField : outputFields){ if((name).equals(outputField.getName())){ return outputField; } } } return null; } }
@Override public Feature apply(Model model){ Output output = model.getOutput(); if(output == null || !output.hasOutputFields()){ throw new IllegalArgumentException(); } OutputField outputField = Iterables.getLast(output.getOutputFields()); return new ContinuousFeature(null, outputField.getName(), outputField.getDataType()); } };
public org.dmg.pmml.Model registerModel(SparkMLEncoder encoder){ Schema schema = encodeSchema(encoder); Label label = schema.getLabel(); org.dmg.pmml.Model model = encodeModel(schema); List<OutputField> sparkOutputFields = registerOutputFields(label, encoder); if(sparkOutputFields != null && sparkOutputFields.size() > 0){ org.dmg.pmml.Model lastModel = getLastModel(model); Output output = ModelUtil.ensureOutput(lastModel); List<OutputField> outputFields = output.getOutputFields(); outputFields.addAll(0, sparkOutputFields); } return model; }
static public Set<Field<?>> getActiveFields(DeepFieldResolver resolver, Model model){ Set<Field<?>> modelFields = getModelFields(resolver, model); FieldReferenceFinder fieldReferenceFinder = new FieldReferenceFinder(){ @Override public VisitorAction visit(LocalTransformations localTransformations){ return VisitorAction.SKIP; } }; fieldReferenceFinder.applyTo(model); Set<Field<?>> activeFields = FieldUtil.selectAll(modelFields, fieldReferenceFinder.getFieldNames()); Output output = model.getOutput(); if(output != null){ activeFields.removeAll(output.getOutputFields()); } return activeFields; }
activeFields.removeAll(output.getOutputFields());
static public Set<Field<?>> getActiveFields(DeepFieldResolver resolver, Model model){ Set<Field<?>> modelFields = getModelFields(resolver, model); FieldReferenceFinder fieldReferenceFinder = new FieldReferenceFinder(){ @Override public VisitorAction visit(LocalTransformations localTransformations){ return VisitorAction.SKIP; } }; fieldReferenceFinder.applyTo(model); Set<Field<?>> activeFields = FieldUtil.selectAll(modelFields, fieldReferenceFinder.getFieldNames()); Output output = model.getOutput(); if(output != null){ activeFields.removeAll(output.getOutputFields()); } return activeFields; }
activeFields.removeAll(output.getOutputFields());
@Override public VisitorAction accept(Visitor visitor) { VisitorAction status = visitor.visit(this); if (status == VisitorAction.CONTINUE) { visitor.pushParent(this); if ((status == VisitorAction.CONTINUE)&&hasExtensions()) { status = org.dmg.pmml.PMMLObject.traverse(visitor, getExtensions()); } if ((status == VisitorAction.CONTINUE)&&hasOutputFields()) { status = org.dmg.pmml.PMMLObject.traverse(visitor, getOutputFields()); } visitor.popParent(); } if (status == VisitorAction.TERMINATE) { return VisitorAction.TERMINATE; } return VisitorAction.CONTINUE; }
@Override public VisitorAction accept(Visitor visitor) { VisitorAction status = visitor.visit(this); if (status == VisitorAction.CONTINUE) { visitor.pushParent(this); if ((status == VisitorAction.CONTINUE)&&hasExtensions()) { status = org.dmg.pmml.PMMLObject.traverse(visitor, getExtensions()); } if ((status == VisitorAction.CONTINUE)&&hasOutputFields()) { status = org.dmg.pmml.PMMLObject.traverse(visitor, getOutputFields()); } visitor.popParent(); } if (status == VisitorAction.TERMINATE) { return VisitorAction.TERMINATE; } return VisitorAction.CONTINUE; }
List<OutputField> outputFields = output.getOutputFields();