naivebayesoperator.java

一个数据挖掘软件ALPHAMINERR的整个过程的JAVA版源代码

/*
 * $Author$
 * $Date$
 * $Revision$
 */
package eti.bi.alphaminer.patch.standard.operation.operator;

import java.util.Vector;

import com.prudsys.pdm.Core.MiningAlgorithm;
import com.prudsys.pdm.Core.MiningAlgorithmSpecification;
import com.prudsys.pdm.Core.MiningAttribute;
import com.prudsys.pdm.Core.MiningException;
import com.prudsys.pdm.Core.MiningModel;
import com.prudsys.pdm.Core.NumericAttribute;
import com.prudsys.pdm.Models.Supervised.SupervisedMiningSettings;
import com.prudsys.pdm.Utils.GeneralUtils;

import eti.bi.alphaminer.core.handler.ICaseHandler;
import eti.bi.alphaminer.operation.operator.INodeInfo;
import eti.bi.alphaminer.operation.operator.ModelOperator;
import eti.bi.alphaminer.operation.operator.Operator;
import eti.bi.common.Locale.Resource;
import eti.bi.alphaminer.vo.BIData;
import eti.bi.alphaminer.vo.BIModel;
import eti.bi.alphaminer.vo.BIObject;
import eti.bi.alphaminer.vo.IBIData;
import eti.bi.alphaminer.vo.IBIModel;
import eti.bi.alphaminer.vo.IOperatorNode;
import eti.bi.exception.AppException;
import eti.bi.exception.SysException;

public class NaiveBayesOperator extends ModelOperator {

	
	/* Parameter name for NavieBayes Operator in BIML */
	public static String KERNAL = "kernal";
	public static String DISCRETIZATION = "discretization";
	
	public static boolean DEFAULT_KERNAL = true;
	public static boolean DEFAULT_DISCRETIZATION = false;
	
	/* Parameter name for MiningSettingSpecification and MiningAlgorithm */
	public static final String ALGORITHM_NAME = "NaiveBayes (Weka)";
	private static String MAP_WEKA_CLASS_PARAMETERS = "wekaClassParameters";
	
	
	/**
	 * @param a_CaseID
	 * @param a_CaseWindow
	 * @param aOperatorInfo
	 */
	public NaiveBayesOperator(String a_CaseID, INodeInfo aNodeInfo, ICaseHandler aCaseHandler) {
		super(a_CaseID, aNodeInfo, aCaseHandler);
		setDefaultModelName("NaiveBayes model");
		//2006/07/29 Xiaojun Chen
		PredictionAssessmentOperator.registerParentsDefinitionID(aNodeInfo.getDefinitionID());
		ScoreOperator.registerParentsDefinitionID(aNodeInfo.getDefinitionID());
	}

	/**
	 * 
	 */
	private static final long serialVersionUID = 1L;

	/**
	 * Set node id and update operator text of the DecisionTreeOperator at the same time.
	 * @param a_NodeID ID of the node
	 */
	public void setNodeID(String a_NodeID) {
		setLabel(getDescription() + " [" + a_NodeID + "]");
		setDefaultModelName(Resource.srcStr("NAVIEBAYES") + a_NodeID);
		super.setNodeID(a_NodeID);
	}
	
	/**
	 * Set node id and update operator text of the DecisionTreeOperator at the same time.
	 * @param a_NodeID ID of the node
	 */
	public void setDescription(String a_Description) {
		m_Description = a_Description;
		setLabel(m_Description + " [" + m_NodeID + "]");
		setDefaultModelName(Resource.srcStr("NAVIEBAYES") + m_NodeID);
	}

	
	@SuppressWarnings("unchecked")
	public void execute(IOperatorNode a_OperatorNode, Vector a_Parents)
			throws SysException, AppException, MiningException {
		
		String navieBayesparameter = "";
		
		/*Get parameter from user input*/
		String kernal = (String) a_OperatorNode.getParameterValue(KERNAL);
		boolean ker = DEFAULT_KERNAL;
		if(kernal!=null){
			ker = new Boolean(kernal).booleanValue();
		}
		if(ker){
			navieBayesparameter = "-K";
		}
		
		
		String discretization = (String) a_OperatorNode.getParameterValue(DISCRETIZATION);
		boolean dis = DEFAULT_DISCRETIZATION;
		if(discretization!=null){
			dis = new Boolean(discretization).booleanValue();
		}
		if(dis&&!ker){
			navieBayesparameter = "-D";
		}
		
		/* Get input bi object from parent node */
		Operator parentOp = (Operator)a_Parents.elementAt(0);
		setInputBIObject(parentOp.getOutputBIObject());
		IBIData aInputBIData = getInputBIObject().getBIData();
		aInputBIData.getMiningStoredData().reset();
		
		/* Prepare output data model */
		BIData aOutputBIData = new BIData(getCaseID(), getNodeID());
		aOutputBIData.setTargetAttribute(aInputBIData.getTargetAttribute());
		aOutputBIData.setTransformActionHistory(aInputBIData.getTransformActionHistory());
		aOutputBIData.setTargetAttribute(aInputBIData.getTargetAttribute());
		aOutputBIData.setMiningStoredData(aInputBIData.getMiningStoredData());
		BIModel aOutputBIModel = new BIModel(getCaseID(), getNodeID(), IBIModel.TYPE_CLASSIFIER);
		
		/* Check attributes */
		MiningAttribute targetAttribute = aInputBIData.getTargetAttribute();
		aOutputBIModel.setTargetAttribute(targetAttribute);
		if (targetAttribute==null)
		{
			throw new AppException("Categorical Target attribute is missing. Please add target attribute by using Data Set Attribute Node.");
		}else if (targetAttribute instanceof NumericAttribute)
		{
			m_SystemMessageHandler.appendMessage("Attribute \""+targetAttribute.getName() + "\" is not Categorical.");
			throw new AppException("Attribute \""+targetAttribute.getName() + "\" should be Categorical.");
		}
		
		/*Create MiningSettings object and assign metadata*/
		SupervisedMiningSettings miningSettings = new SupervisedMiningSettings();
		miningSettings.setDataSpecification(aInputBIData.getMetaData());
		
		 /* Assign settings */
	    miningSettings.setTarget(targetAttribute);
	    try {
	    	miningSettings.verifySettings();
	    } catch (Exception e)
		{
	    	m_SystemMessageHandler.appendMessage("Invalid parameters in building the Decision Tree model.");
	    	throw new AppException("Invalid parameters in building the Decision Tree model.");
	    }	    
		
	    /* Set MiningSettings */
		aOutputBIModel.setMiningSettings(miningSettings);
		/* Get default mining algorithm specification from 'algorithms.xml' */
	    MiningAlgorithmSpecification miningAlgorithmSpecification =
	    MiningAlgorithmSpecification.getMiningAlgorithmSpecification( ALGORITHM_NAME, getNodeInfo());
	    
	    if( miningAlgorithmSpecification == null )
		      throw new MiningException( "Can't find NaiveBayes classification method." );
	    
	    /* Get class name from algorithms specification */
	    String className = miningAlgorithmSpecification.getClassname();
	    if( className == null )
	      throw new MiningException( "className attribute expected." );
	    
	    /* Set MiningAlgorithmSpecification */
		miningAlgorithmSpecification.setMAPValue(MAP_WEKA_CLASS_PARAMETERS, navieBayesparameter);
		aOutputBIModel.setMiningAlgorithmSpecification(miningAlgorithmSpecification);
		displayMiningAlgSpecParameters(miningAlgorithmSpecification);
	    
		 /* Set and display mining parameters */
	    GeneralUtils.displayMiningAlgSpecParameters(miningAlgorithmSpecification);
	    
	    /* Create algorithm object with default values */
	    MiningAlgorithm algorithm = GeneralUtils.createMiningAlgorithmInstance(className, this.getClass().getClassLoader());
						
		algorithm.setMiningInputStream(aInputBIData.getMiningStoredData());
		algorithm.setMiningSettings(miningSettings);
		algorithm.setMiningAlgorithmSpecification(miningAlgorithmSpecification);
		try
		{
			algorithm.verify();
		} catch(IllegalArgumentException e)
		{
			throw new MiningException(e.getMessage());
		}
		
		MiningModel model = algorithm.buildModel();
		
		m_SystemMessageHandler.appendMessage(Resource.srcStr("calculationtime")+" [s]: " + algorithm.getTimeSpentToBuildModel()+Resource.srcStr("ms"));
		m_SystemMessageHandler.nextLine();
		
		aOutputBIModel.setMiningModel(model);
		aOutputBIModel.setModelName(m_DefaultModelName);
		
		m_OutputBIObject.setBIData(aOutputBIData);	
		m_OutputBIObject.setBIModel(aOutputBIModel);
		
		//a_OperatorNode.setParameterValue("Temporary model", aOutputBIModel.getTempBIModelPath());		
		
		//aOutputBIModel.writeTempBIModel();
		
	}

	/**
	 * Test if the NavieBayes Operator contains any results.
	 * @return true if NavieBayes Operator has result; false otherwise.
	 */
	public boolean hasResult() throws SysException {
		
		if (m_OutputBIObject != null)
		{
			return (m_OutputBIObject.hasResult(BIObject.DATA) &&
					m_OutputBIObject.hasResult(BIObject.MODEL));
		}else
		{
			return false;
		}
		
	}
}