aima.core.learning.neural.IrisNNDataSet Maven / Gradle / Ivy
package aima.core.learning.neural;
import java.util.ArrayList;
/**
* @author Ravi Mohan
*
*/
public class IrisNNDataSet extends NNDataSet {
@Override
public void setTargetColumns() {
// assumed that data from file has been pre processed
// TODO this should be
// somewhere else,in the
// super class.
// Type != class Aargh! I want more
// powerful type systems
targetColumnNumbers = new ArrayList();
int size = nds.get(0).size();
targetColumnNumbers.add(size - 1); // last column
targetColumnNumbers.add(size - 2); // last but one column
targetColumnNumbers.add(size - 3); // and the one before that
}
}