public PredictionModelMetaData(
Class<? extends PredictionModel> modelClass, ExampleSetMetaData trainingSetMetaData) {
super(modelClass, trainingSetMetaData);
if (trainingSetMetaData != null) {
AttributeMetaData labelAttributeMetaData = trainingSetMetaData.getLabelMetaData();
if (labelAttributeMetaData != null) {
this.predictedLabelMetaData = labelAttributeMetaData.copy();
this.predictedLabelMetaData.setRole(Attributes.PREDICTION_NAME);
this.predictedLabelMetaData.setName("prediction(" + predictedLabelMetaData.getName() + ")");
if (predictedLabelMetaData.isNumerical()) {
this.predictedLabelMetaData.setValueSetRelation(SetRelation.SUPERSET);
}
this.predictedLabelMetaData.setMean(new MDReal());
// creating confidence attributes
generatedPredictionAttributes.add(predictedLabelMetaData);
if (predictedLabelMetaData.isNominal()) {
if (predictedLabelMetaData.getValueSet().isEmpty()) {
AttributeMetaData confidence =
new AttributeMetaData(
Attributes.CONFIDENCE_NAME + "(?)",
Ontology.REAL,
Attributes.CONFIDENCE_NAME + "_" + "?");
confidence.setValueRange(new Range(0, 1), SetRelation.SUBSET);
generatedPredictionAttributes.add(confidence);
predictedLabelMetaData.setValueSetRelation(SetRelation.SUPERSET);
} else {
for (String value : predictedLabelMetaData.getValueSet()) {
AttributeMetaData confidence =
new AttributeMetaData(
Attributes.CONFIDENCE_NAME + "(" + value + ")",
Ontology.REAL,
Attributes.CONFIDENCE_NAME + "_" + value);
confidence.setValueRange(new Range(0, 1), SetRelation.SUBSET);
generatedPredictionAttributes.add(confidence);
}
}
}
}
}
}
