transtype.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.
 */

 /**
  * Title: XELOPES Data Mining Library
  * Description: The XELOPES library is an open platform-independent and data-source-independent library for Embedded Data Mining.
  * Copyright: Copyright (c) 2002 Prudential Systems Software GmbH
  * Company: ZSoft (www.zsoft.ru), Prudsys (www.prudsys.com)
  * @author Michael Thess
  * @version 1.1
  */
package com.prudsys.pdm.Transform.OneToOne;

import com.prudsys.pdm.Core.MiningAttribute;
import com.prudsys.pdm.Core.MiningException;
import com.prudsys.pdm.Core.NumericAttribute;
import com.prudsys.pdm.Transform.OneToOneMapping;

/**
 * Transforms an attribute to target data type. This allows to change
 * the data type of an attribute.
 *
 * Transforms a numeric into a numeric attribute using TransTypeNum
 * and a categorical into a categorical attribute using TransTypeCateg.
 *
 */
public class TransType extends OneToOneMapping
{
  // -----------------------------------------------------------------------
  //  Variables declarations
  // -----------------------------------------------------------------------
  /** Mapping object TransTypeNum or TransTypeCateg. */
  private OneToOneMapping transMap = null;

  /** Data type of target attribute. */
  private int targetDataType = MiningAttribute.DEFAULT;

  // -----------------------------------------------------------------------
  //  Constructor
  // -----------------------------------------------------------------------
  /**
   * Empty constructor.
   */
  public TransType()
  {
  }

  // -----------------------------------------------------------------------
  //  Getter and setter methods
  // -----------------------------------------------------------------------
  /**
   * Returns target attribute data type.
   *
   * @return target attribute data type
   */
  public int getTargetDataType() {

    return targetDataType;
  }

  /**
   * Sets target attribute data type.
   *
   * @param targetDataType new data type of target attribute
   */
  public void setTargetDataType(int targetDataType) {

    this.targetDataType = targetDataType;
  }

  // -----------------------------------------------------------------------
  //  Transformation methods
  // -----------------------------------------------------------------------
  /**
   * Transforms the source attribute. The result is the
   * target attribute with new data type.
   *
   * @return transformed attribute
   * @exception MiningException could not transform attribute
   */
  public MiningAttribute transformAttribute() throws MiningException
  {
    MiningAttribute sourceAtt = getSourceAttribute();

    if (sourceAtt == null)
      throw new MiningException("Could not find source attribute: " + sourceName);

    if ( sourceAtt instanceof NumericAttribute ) {
      transMap = new TransTypeNum();
      ( (TransTypeNum) transMap ).setTargetDataType(targetDataType);
    }
    else {
      transMap = new TransTypeCateg();
      ( (TransTypeCateg) transMap ).setTargetDataType(targetDataType);
    }
    transMap.setSourceAttribute(sourceAtt);

    return transMap.transformAttribute();
  }

  /**
   * Transforms value of source attribute into the value of new data type.
   *
   * @param attributeValue value to be transformed
   * @return transformed value
   * @exception MiningException could not transform attribute value
   */
  public double transformAttributeValue( double attributeValue ) throws MiningException
  {
    return transMap.transformAttributeValue(attributeValue);
  }

  // -----------------------------------------------------------------------
  //  Methods of PMML handling
  // -----------------------------------------------------------------------
  /**
   * Creates PMML object DerivedField of this object. Not supported.
   *
   * @return DerivedField element
   * @see com.prudsys.pdm.Adapters.PmmlVersion20.TransformationDictionary
   * @exception MiningException could not create PMML object
   */
  public Object createPmmlObject() throws MiningException
  {
    return transMap.createPmmlObject();
  }

  /**
   * Creates this object from PMML. Not supported.
   *
   * @param pmml pmml element
   * @exception MiningException always thrown
   */
  public void parsePmmlObject(Object pmml) throws MiningException
  {
    transMap.parsePmmlObject(pmml);
  }
}