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

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.

The newest version!

package aima.core.learning.neural;

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

import aima.core.learning.framework.Example;
import aima.core.util.datastructure.Pair;

/**
 * @author Ravi Mohan
 * 
 */
public class IrisDataSetNumerizer implements Numerizer {

	public Pair, List> numerize(Example e) {
		List input = new ArrayList();
		List desiredOutput = new ArrayList();

		double sepal_length = e.getAttributeValueAsDouble("sepal_length");
		double sepal_width = e.getAttributeValueAsDouble("sepal_width");
		double petal_length = e.getAttributeValueAsDouble("petal_length");
		double petal_width = e.getAttributeValueAsDouble("petal_width");

		input.add(sepal_length);
		input.add(sepal_width);
		input.add(petal_length);
		input.add(petal_width);

		String plant_category_string = e
				.getAttributeValueAsString("plant_category");

		desiredOutput = convertCategoryToListOfDoubles(plant_category_string);

		Pair, List> io = new Pair, List>(
				input, desiredOutput);

		return io;
	}

	public String denumerize(List outputValue) {
		List rounded = new ArrayList();
		for (Double d : outputValue) {
			rounded.add(round(d));
		}
		if (rounded.equals(Arrays.asList(0.0, 0.0, 1.0))) {
			return "setosa";
		} else if (rounded.equals(Arrays.asList(0.0, 1.0, 0.0))) {
			return "versicolor";
		} else if (rounded.equals(Arrays.asList(1.0, 0.0, 0.0))) {
			return "virginica";
		} else {
			return "unknown";
		}
	}

	//
	// PRIVATE METHODS
	//
	private double round(Double d) {
		if (d < 0) {
			return 0.0;
		}
		if (d > 1) {
			return 1.0;
		} else {
			return Math.round(d);
		}
	}

	private List convertCategoryToListOfDoubles(
			String plant_category_string) {
		if (plant_category_string.equals("setosa")) {
			return Arrays.asList(0.0, 0.0, 1.0);
		} else if (plant_category_string.equals("versicolor")) {
			return Arrays.asList(0.0, 1.0, 0.0);
		} else if (plant_category_string.equals("virginica")) {
			return Arrays.asList(1.0, 0.0, 0.0);
		} else {
			throw new RuntimeException("invalid plant category");
		}
	}
}