aima.core.learning.neural.IrisNNDataSet Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of aima-core Show documentation

AIMA-Java Core Algorithms from the book Artificial Intelligence a Modern Approach 3rd Ed.

There is a newer version: 3.0.0

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
	}
}