DatumResult.java

package com.datascience.core.results;

import java.util.HashMap;
import java.util.Map;

/**
 * User: artur
 */
public class DatumResult {
	protected Map<String, Double> categoryProbabilites;

	public DatumResult(){
		categoryProbabilites = new HashMap<String, Double>();
	}

	public Map<String, Double> getCategoryProbabilites(){
		return categoryProbabilites;
	}

	public Double getCategoryProbability(String categoryName){
		return categoryProbabilites.get(categoryName);
	}

	public void setCategoryProbabilites(Map<String, Double> cp){
		categoryProbabilites = cp;
	}

}