Java TreeModel.getMiningFunction方法代码示例

import org.dmg.pmml.tree.TreeModel; //导入方法依赖的package包/类
@Override
public VisitorAction visit(Node node){
	TreeModel treeModel = getTreeModel();

	MiningFunction miningFunction = treeModel.getMiningFunction();
	switch(miningFunction){
		case REGRESSION:
			if(node.hasScoreDistributions()){
				List<ScoreDistribution> scoreDistributions = node.getScoreDistributions();

				scoreDistributions.clear();
			}
			break;
		default:
			break;
	}

	return super.visit(node);
}
 

import org.dmg.pmml.tree.TreeModel; //导入方法依赖的package包/类
@Override
public VisitorAction visit(TreeModel treeModel){
	TreeModel.MissingValueStrategy missingValueStrategy = treeModel.getMissingValueStrategy();
	TreeModel.NoTrueChildStrategy noTrueChildStrategy = treeModel.getNoTrueChildStrategy();
	TreeModel.SplitCharacteristic splitCharacteristic = treeModel.getSplitCharacteristic();

	if(!(TreeModel.MissingValueStrategy.NONE).equals(missingValueStrategy) || !(TreeModel.NoTrueChildStrategy.RETURN_NULL_PREDICTION).equals(noTrueChildStrategy) || !(TreeModel.SplitCharacteristic.BINARY_SPLIT).equals(splitCharacteristic)){
		throw new IllegalArgumentException();
	}

	treeModel
		.setMissingValueStrategy(TreeModel.MissingValueStrategy.NULL_PREDICTION)
		.setSplitCharacteristic(TreeModel.SplitCharacteristic.MULTI_SPLIT);

	MiningFunction miningFunction = treeModel.getMiningFunction();
	switch(miningFunction){
		case REGRESSION:
			treeModel.setNoTrueChildStrategy(TreeModel.NoTrueChildStrategy.RETURN_LAST_PREDICTION);
			break;
		case CLASSIFICATION:
			break;
		default:
			throw new IllegalArgumentException();
	}

	return super.visit(treeModel);
}
 

import org.dmg.pmml.tree.TreeModel; //导入方法依赖的package包/类
@Override
public Map<FieldName, ?> evaluate(ModelEvaluationContext context){
	TreeModel treeModel = ensureScorableModel();

	ValueFactory<?> valueFactory;

	MathContext mathContext = treeModel.getMathContext();
	switch(mathContext){
		case FLOAT:
		case DOUBLE:
			valueFactory = getValueFactory();
			break;
		default:
			throw new UnsupportedAttributeException(treeModel, mathContext);
	}

	Map<FieldName, ?> predictions;

	MiningFunction miningFunction = treeModel.getMiningFunction();
	switch(miningFunction){
		case REGRESSION:
			predictions = evaluateRegression(valueFactory, context);
			break;
		case CLASSIFICATION:
			predictions = evaluateClassification(valueFactory, context);
			break;
		case ASSOCIATION_RULES:
		case SEQUENCES:
		case CLUSTERING:
		case TIME_SERIES:
		case MIXED:
			throw new InvalidAttributeException(treeModel, miningFunction);
		default:
			throw new UnsupportedAttributeException(treeModel, miningFunction);
	}

	return OutputUtil.evaluate(predictions, context);
}
 

import org.dmg.pmml.tree.TreeModel; //导入方法依赖的package包/类
private void handleTreeModelPush(TreeModel treeModel){
	this.miningFunction = treeModel.getMiningFunction();

	this.categoricalFields.clear();
}
 

import org.dmg.pmml.tree.TreeModel; //导入方法依赖的package包/类
private void handleTreeModelPush(TreeModel treeModel){
	this.miningFunction = treeModel.getMiningFunction();
}