Java Entity类代码示例

import org.dmg.pmml.Entity; //导入依赖的package包/类
static
public NeuralOutputs createRegressionNeuralOutputs(List<? extends Entity> entities, ContinuousLabel continuousLabel){

	if(entities.size() != 1){
		throw new IllegalArgumentException();
	}

	Entity entity = Iterables.getOnlyElement(entities);

	DerivedField derivedField = new DerivedField(OpType.CONTINUOUS, continuousLabel.getDataType())
		.setExpression(new FieldRef(continuousLabel.getName()));

	NeuralOutput neuralOutput = new NeuralOutput()
		.setOutputNeuron(entity.getId())
		.setDerivedField(derivedField);

	NeuralOutputs neuralOutputs = new NeuralOutputs()
		.addNeuralOutputs(neuralOutput);

	return neuralOutputs;
}
 

import org.dmg.pmml.Entity; //导入依赖的package包/类
static
public NeuralOutputs createClassificationNeuralOutputs(List<? extends Entity> entities, CategoricalLabel categoricalLabel){

	if(entities.size() != categoricalLabel.size()){
		throw new IllegalArgumentException();
	}

	NeuralOutputs neuralOutputs = new NeuralOutputs();

	for(int i = 0; i < categoricalLabel.size(); i++){
		Entity entity = entities.get(i);

		DerivedField derivedField = new DerivedField(OpType.CATEGORICAL, categoricalLabel.getDataType())
			.setExpression(new NormDiscrete(categoricalLabel.getName(), categoricalLabel.getValue(i)));

		NeuralOutput neuralOutput = new NeuralOutput()
			.setOutputNeuron(entity.getId())
			.setDerivedField(derivedField);

		neuralOutputs.addNeuralOutputs(neuralOutput);
	}

	return neuralOutputs;
}
 

import org.dmg.pmml.Entity; //导入依赖的package包/类
@Override
public BiMap<String, Entity> load(NeuralNetwork neuralNetwork){
	ImmutableBiMap.Builder<String, Entity> builder = new ImmutableBiMap.Builder<>();

	AtomicInteger index = new AtomicInteger(1);

	NeuralInputs neuralInputs = neuralNetwork.getNeuralInputs();
	for(NeuralInput neuralInput : neuralInputs){
		builder = EntityUtil.put(neuralInput, index, builder);
	}

	List<NeuralLayer> neuralLayers = neuralNetwork.getNeuralLayers();
	for(NeuralLayer neuralLayer : neuralLayers){
		List<Neuron> neurons = neuralLayer.getNeurons();

		for(int i = 0; i < neurons.size(); i++){
			Neuron neuron = neurons.get(i);

			builder = EntityUtil.put(neuron, index, builder);
		}
	}

	return builder.build();
}
 

import org.dmg.pmml.Entity; //导入依赖的package包/类
static
public List<OutputField> createAffinityFields(final DataType dataType, List<? extends Entity> entities){
	Function<Entity, OutputField> function = new Function<Entity, OutputField>(){

		@Override
		public OutputField apply(Entity entity){
			return createAffinityField(dataType, entity.getId());
		}
	};

	return Lists.newArrayList(Lists.transform(entities, function));
}
 

import org.dmg.pmml.Entity; //导入依赖的package包/类
static
public Neuron createNeuron(List<? extends Entity> entities, List<Double> weights, Double bias){

	if(entities.size() != weights.size()){
		throw new IllegalArgumentException();
	}

	Neuron neuron = new Neuron();

	for(int i = 0; i < entities.size(); i++){
		Entity entity = entities.get(i);
		Double weight = weights.get(i);

		if(weight.isNaN() || ValueUtil.isZero(weight)){
			continue;
		}

		Connection connection = new Connection()
			.setFrom(entity.getId())
			.setWeight(weight);

		neuron.addConnections(connection);
	}

	if(!bias.isNaN() && !ValueUtil.isZero(bias)){
		neuron.setBias(bias);
	}

	return neuron;
}
 

import org.dmg.pmml.Entity; //导入依赖的package包/类
@Override
public BiMap<String, Entity> getEntityRegistry(){

	if(this.entityRegistry == null){
		this.entityRegistry = getValue(NeuralNetworkEvaluator.entityCache);
	}

	return this.entityRegistry;
}
 

import org.dmg.pmml.Entity; //导入依赖的package包/类
static
public <E extends Entity> String getId(E entity, BiMap<String, E> entityRegistry){
	String id = entity.getId();

	if(id == null){
		BiMap<E, String> inversedEntityRegistry = entityRegistry.inverse();

		return inversedEntityRegistry.get(entity);
	}

	return id;
}
 

import org.dmg.pmml.Entity; //导入依赖的package包/类
static
public <E extends Entity> ImmutableBiMap<String, E> buildBiMap(List<E> entities){
	ImmutableBiMap.Builder<String, E> builder = new ImmutableBiMap.Builder<>();

	builder = putAll(entities, new AtomicInteger(1), builder);

	return builder.build();
}
 

import org.dmg.pmml.Entity; //导入依赖的package包/类
static
public <E extends Entity> ImmutableBiMap.Builder<String, E> put(E entity, AtomicInteger index, ImmutableBiMap.Builder<String, E> builder){
	String implicitId = String.valueOf(index.getAndIncrement());

	String id = entity.getId();
	if(id == null){
		id = implicitId;
	}

	return builder.put(id, entity);
}
 

import org.dmg.pmml.Entity; //导入依赖的package包/类
static
public <E extends Entity> ImmutableBiMap.Builder<String, E> putAll(List<E> entities, AtomicInteger index, ImmutableBiMap.Builder<String, E> builder){

	for(E entity : entities){
		builder = put(entity, index, builder);
	}

	return builder;
}