categoricaltransformationoperator.java

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

/*
 *    This program is free software; you can redistribute it and/or modify
 *    it under the terms of the GNU General Public License as published by
 *    the Free Software Foundation; either version 2 of the License, or
 *    (at your option) any later version.
 *
 *    This program is distributed in the hope that it will be useful,
 *    but WITHOUT ANY WARRANTY; without even the implied warranty of
 *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *    GNU General Public License for more details.
 *
 *    You should have received a copy of the GNU General Public License
 *    along with this program; if not, write to the Free Software
 *    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 */

package eti.bi.alphaminer.patch.standard.operation.operator;


import java.util.ArrayList;
import java.util.Vector;



import com.prudsys.pdm.Core.CategoricalAttribute;
import com.prudsys.pdm.Core.Category;
import com.prudsys.pdm.Core.MiningAttribute;
import com.prudsys.pdm.Core.MiningDataSpecification;
import com.prudsys.pdm.Core.MiningException;
import com.prudsys.pdm.Core.NumericAttribute;
import com.prudsys.pdm.Input.MiningStoredData;
import com.prudsys.pdm.Transform.MiningTransformationFactory;
import com.prudsys.pdm.Transform.MiningTransformationStep;
import com.prudsys.pdm.Transform.OneToOne.CategMapping;
import com.prudsys.pdm.Transform.OneToOne.CategNumMapping;
import com.prudsys.pdm.Transform.OneToOne.Numerization;

import eti.bi.alphaminer.core.handler.ICaseHandler;
import eti.bi.alphaminer.core.transform.XelopesTransformAction;
import eti.bi.alphaminer.operation.operator.INodeInfo;
import eti.bi.alphaminer.operation.operator.Operator;
import eti.bi.alphaminer.operation.operator.TransformOperator;
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;
import eti.bi.util.ValueValidator;
import eti.bi.common.Locale.Resource;

/**
 * ReplacementOperator is a kind of Operator
 */
public class CategoricalTransformationOperator extends TransformOperator {

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


	/**
	 * @param a_CaseID
	 * @param a_CaseWindow
	 * @param aOperatorInfo
	 */
	public CategoricalTransformationOperator(String a_CaseID, INodeInfo aNodeInfo, ICaseHandler aCaseHandler) {
		super(a_CaseID, aNodeInfo, aCaseHandler);
		// TODO Auto-generated constructor stub
	}


	public final static int CATEG_MAPPING = 0;
    public final static int CATEG_NUM_MAPPING = 1;
    public final static int NUMERIZATION = 2;
    @SuppressWarnings("unused")
	private final static String CATEG = "Categorical";
    @SuppressWarnings("unused")
	private final static String NUMERIC = "Numeric";
        

	/**
	 * Set node id and update operator text of the Categorical Transformation at the same time.
	 * @param a_NodeID ID of the node
	 */
	public void setNodeID(String a_NodeID) {
		setLabel(getDescription() + " [" + a_NodeID + "]");
		super.setNodeID(a_NodeID);
	}
	
	/**
	 * Set node id and update operator text of the Categorical Transformation 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 + "]");		
	}
	
	/* (non-Javadoc)
	 * @see eti.bi.alphaminer.ui.operator.Operator#hasResult()
	 */
	public boolean hasResult() {
	    /* Normally the operator has to check if the output data object and/or the output
   		 * model object are exist by calling m_OutputBIObject.hasResult(int) by passing BIObject.DATA
   		 * and BIObject.MODEL respectively. You may check both existance or either one by simple modifying
   		 * the logic below.  
   		 */
   		if (m_OutputBIObject != null)
   		{
   			return (m_OutputBIObject.hasResult(BIObject.DATA)); 
   		}else
   		{
   			return false;
   		}
	}

	/* (non-Javadoc)
	 * @see eti.bi.alphaminer.ui.operator.Operator#execute(eti.bi.alphaminer.vo.OperatorNode, java.util.Vector)
	 */
	public void execute(IOperatorNode a_OperatorNode, Vector a_Parents) throws MiningException, AppException, SysException {
	    /* Get input bi object from parent node */
   		Operator parentOp = (Operator)a_Parents.elementAt(0);
   		setInputBIObject(parentOp.getOutputBIObject());
   		IBIData aInputBIData = getInputBIObject().getBIData();
   		
   		
   		/* Get parameter from user input */
   		validateParameters(aInputBIData.getMetaData(),a_OperatorNode);
//   		String aTargetAttrName = a_OperatorNode.getParameterValue("target");
	    int mode = Integer.valueOf((String) a_OperatorNode.getParameterValue("mode")).intValue();
	   
	    
	    /* Prepare output mining data */
   		IBIData aOutputBIData = new BIData(getCaseID(), getNodeID());
   		@SuppressWarnings("unused") IBIModel aOutputBIModel = new BIModel(getCaseID(), getNodeID(), IBIModel.TYPE_CLASSIFIER);
   		
   		
   		
   		/* Execute transform */
   		MiningTransformationFactory mtf = new MiningTransformationFactory();
   		
   		switch(mode){
   		    case CATEG_MAPPING:
	   		    CategMapping categMapping= preparecCategMapping(aInputBIData.getMetaData(),a_OperatorNode);
	   			mtf.addOneToOneMapping(categMapping);
	   			
   		        break;
   		    case CATEG_NUM_MAPPING:
   		        CategNumMapping categNumMapping= preparecCategNumMapping(aInputBIData.getMetaData(),a_OperatorNode);
	   			mtf.addOneToOneMapping(categNumMapping);
   		        break;
   		    case NUMERIZATION:
   		        Numerization numerization= preparecNumerization(aInputBIData.getMetaData(),a_OperatorNode);
	   			mtf.addOneToOneMapping(numerization);
   		        break;
   		        
   		};
   		
   		MiningTransformationStep mts = mtf.createMiningTransformationStep();
		XelopesTransformAction aTransformAction = new XelopesTransformAction(m_CaseID, m_NodeID, mts);
		MiningStoredData aOutputMiningStoredData = aTransformAction.transform(aInputBIData.getMiningStoredData());
   	 		
		
		
		/* Set Output Mining Data */
		aOutputBIData.setMiningStoredData(aOutputMiningStoredData);
		aOutputBIData.copyTransformActionHistory(aInputBIData.getTransformActionHistory());
		aOutputBIData.addTransformActionHistory(aTransformAction);
//		MiningAttribute aTargetAttribute = (MiningAttribute) aOutputBIData.getMetaData().getMiningAttribute(aTargetAttrName);
//		aOutputBIData.setTargetAttribute(aTargetAttribute);
		aOutputBIData.copyTargetAttribute(aInputBIData.getTargetAttribute());
		m_OutputBIObject.setBIData(aOutputBIData);		
		
		/* set run time parameter value to the node object (It needs to be stored in the BIML) */
		//a_OperatorNode.setParameterValue("Temporary data", aOutputBIData.getTempBIDataPath());				
		
		/* write temp data */
		//aOutputBIData.writeTempBIData();   		
   	}
		
	
	public CategMapping preparecCategMapping(MiningDataSpecification a_MetaData, IOperatorNode a_Node) throws MiningException
	{
	    CategMapping mapping = new CategMapping();
		
		
		String sourceName = null;
		String dataTypeStr = null;
		int dataType = CategoricalAttribute.STRING;
		String targetCategStr = null;
		Vector targetCateg = new Vector();
		//get the target attribute
	    sourceName = (String) a_Node.getParameterValue("target");
	  	    
	    //get the target data type
	    dataTypeStr = (String) a_Node.getParameterValue("targetDataType");
	  	    
	    //get the target categories string
	    targetCategStr = (String) a_Node.getParameterValue("targetCateg");

	
	    MiningAttribute mAtt = a_MetaData.getMiningAttribute(sourceName);
	    dataType = Integer.valueOf(dataTypeStr).intValue();
	    targetCateg = ValueValidator.unescapeParam(targetCategStr);
	   
	    mapping.setSourceName(sourceName);
	    mapping.setTargetName(sourceName);
	    
        for(int i=0; i<((CategoricalAttribute)mAtt).getCategoriesNumber(); i++){
            Category cat = ((CategoricalAttribute)mAtt).getCategory(i);
            String newCat = (String)targetCateg.get(i);
//            System.out.println(newCat);
            switch(dataType){
            	case CategoricalAttribute.STRING: 
            	    if(newCat.trim().length()<=0){
//            	        mapping.addCategoryPair(cat, null);
            	    }else{
            	        mapping.addCategoryPair(cat, new Category(newCat));
            	    }
            		break;
//            	case CategoricalAttribute.DOUBLE:
//            	    mapping.addCategoryPair(cat, new Category(new Double(newCat)));
//        			break;
//        		case CategoricalAttribute.FLOAT:
//        		    mapping.addCategoryPair(cat, new Category(new Float(newCat)));
//    				break;
//    			case CategoricalAttribute.INTEGER: