com.bnd.math.domain.learning.TrainingSet Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of bnd-math Show documentation

The newest version!

package com.bnd.math.domain.learning;

import java.util.ArrayList;
import java.util.List;

/**
 * @author © Peter Banda
 * @since 2012
 */
public class TrainingSet {

	private List> trainingPairs = new ArrayList>();

	public TrainingSet() {
		// nothing to do
	}

	public List> getTrainingPairs() {
		return trainingPairs;
	}

	public void ListTrainingPairs(List> trainingPairs) {
		this.trainingPairs = trainingPairs;
	}

	public void addTrainingPair(TrainingPair trainingPair) {
		trainingPairs.add(trainingPair);
	}

	public void removeTrainingPair(TrainingPair trainingPair) {
		trainingPairs.remove(trainingPair);
	}

	public int getInputArity() {
		return !trainingPairs.isEmpty() ? trainingPairs.get(0).getInput().size() : 0;
	}

	public int getDesiredOutputArity() {
		return !trainingPairs.isEmpty() ? trainingPairs.get(0).getDesiredOutput().size() : 0;
	}
}