customersequentialapply.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 Carsten Weisse
  * @author Michael Thess
  * @version 1.0
  */
package com.prudsys.pdm.Examples;

import java.io.FileReader;

import com.prudsys.pdm.Core.CategoricalAttribute;
import com.prudsys.pdm.Core.Category;
import com.prudsys.pdm.Core.MiningModel;
import com.prudsys.pdm.Models.AssociationRules.ItemSet;
import com.prudsys.pdm.Models.CustomerSeq.CustomRuleSetSeq;
import com.prudsys.pdm.Models.CustomerSeq.CustomSequence;
import com.prudsys.pdm.Models.CustomerSeq.CustomerSequentialMiningModel;
import com.prudsys.pdm.Models.CustomerSeq.CustomerSequentialSettings;

/**
 * Applies a sequential basket analysis model from the PMML file
 * 'CustomerSequentialModel.xml' to a data set.
 */
public class CustomerSequentialApply extends BasisExample {

  /**
   * Empty constructor.
   */
  public CustomerSequentialApply() {
  }

  /**
   * Run the example of this class.
   *
   * @throws Exception error while example is running
   */
  public void runExample() throws Exception {

    // Read sequence rules from PMML file:
    MiningModel model = new CustomerSequentialMiningModel();
    FileReader reader = new FileReader("data/pmml/CustomerSequentialModel.xml");
    model.readPmml(reader);
    CategoricalAttribute categoryItemId = (CategoricalAttribute)
          ((CustomerSequentialSettings) model.getMiningSettings()).getItemId();
    System.out.println("-------------> PMML model read successfully");

    // Build recommendations:
    ((CustomerSequentialMiningModel) model).buildRecommendations();

    // Apply rules to trivial 'customer sequence' ("item_2", "item_3", "item_5"):
    ItemSet is = new ItemSet();
    is.addItem( (int) categoryItemId.getKey( new Category("item_2")) );
    is.addItem( (int) categoryItemId.getKey( new Category("item_3") ));
    is.addItem( (int) categoryItemId.getKey( new Category("item_5") ));
    CustomSequence cs = new CustomSequence();
    cs.addItemSet(is);
    CustomRuleSetSeq crs = (CustomRuleSetSeq) ((CustomerSequentialMiningModel) model).applyModel(cs);
    if (crs != null) {
      System.out.println("ncseqs = " + crs.getSize());
      System.out.println("prem: " + crs.getPremise().toString());
      System.out.println("conc: " + crs.getConclusion().toString());
      System.out.println("Recommendation as string: ");
      for (int i = 0; i < crs.getPremise().getSize(); i++) {
        ItemSet iset = crs.getPremise().getItemSet(i);
        for (int j = 0; j < iset.getSize(); j++) {
          Category recItem = categoryItemId.getCategory(iset.getItemAt(j));
          System.out.print( recItem.getDisplayValue() );
          if (j < iset.getSize() - 1)
            System.out.print(", ");
        };
        if (i < crs.getPremise().getSize() - 1)
          System.out.print("| ");
      };
      System.out.print(" => ");
      for (int i = 0; i < crs.getConclusion().getSize(); i++) {
        ItemSet iset = crs.getConclusion().getItemSet(i);
        for (int j = 0; j < iset.getSize(); j++) {
          Category recItem = categoryItemId.getCategory( iset.getItemAt(j) );
          System.out.print( recItem.getDisplayValue() );
          if (j < iset.getSize() - 1)
            System.out.print(", ");
        };
        if (i < crs.getConclusion().getSize() - 1)
          System.out.print("| ");
        else
          System.out.println();
      };
    }
    else
      System.out.println("No recommendations found.");
  }

  /**
   * Example of applying customer sequence rules to new data.
   *
   * @param args arguments (ignored)
   */
  public static void main(String[] args) {

    try {
      new CustomerSequentialApply().runExample();
    }
    catch (Exception ex) {
      ex.printStackTrace();
    }
  }
}