testmultinomialpotential.java

常用机器学习算法,java编写源代码,内含常用分类算法,包括说明文档

/* Copyright (C) 2003 Univ. of Massachusetts Amherst, Computer Science Dept.
   This file is part of "MALLET" (MAchine Learning for LanguagE Toolkit).
   http://www.cs.umass.edu/~mccallum/mallet
   This software is provided under the terms of the Common Public License,
   version 1.0, as published by http://www.opensource.org.  For further
   information, see the file `LICENSE' included with this distribution. */
package edu.umass.cs.mallet.grmm.test;

import junit.framework.*;
import edu.umass.cs.mallet.grmm.Variable;
import edu.umass.cs.mallet.grmm.MultinomialPotential;
import edu.umass.cs.mallet.grmm.DiscretePotential;

/**
 * Created: Aug 17, 2004
 *
 * @author <A HREF="mailto:[e-mail]>[e-mail]</A>
 * @version $Id: TestMultinomialPotential.java,v 1.1 2004/08/17 20:52:19 casutton Exp $
 */
public class TestMultinomialPotential extends TestCase {

  public TestMultinomialPotential (String name)
  {
    super (name);
  }

  public void testMultiplyMultiplyBy ()
  {
    Variable var = new Variable (4);
    double[] vals = new double[]{ 2.0, 4.0, 6.0, 8.0 };
    double[] vals2 = new double [] { 0.5, 0.5, 0.5, 0.5 };

    double[] vals3 = new double [] { 1, 2, 3, 4, };
    MultinomialPotential ans = new MultinomialPotential (var, vals3);

    MultinomialPotential ptl1 = new MultinomialPotential (var, vals);
    MultinomialPotential ptl2 = new MultinomialPotential (var, vals2);
    DiscretePotential ptl3 = ptl1.multiply (ptl2);
    ptl1.multiplyBy (ptl2);

    assertTrue (ans.almostEquals (ptl1));
    assertTrue (ans.almostEquals (ptl3));
  }

  public void testMultiplyByLogSpace ()
  {
    Variable var = new Variable (4);
    double[] vals = new double[]{ 2.0, 4.0, 6.0, 8.0 };
    double[] vals2 = new double [] { 0.5, 0.5, 0.5, 0.5 };

    double[] vals3 = new double [] { 1, 2, 3, 4, };
    MultinomialPotential ans = new MultinomialPotential (var, vals3);

    MultinomialPotential ptl1 = new MultinomialPotential (var, vals);
    MultinomialPotential ptl2 = new MultinomialPotential (var, vals2);
    ptl1.multiplyBy (ptl2);
    assertTrue (ans.almostEquals (ptl1));

    MultinomialPotential ptl3 = new MultinomialPotential (var, vals);
    MultinomialPotential ptl4 = new MultinomialPotential (var, vals2);
    ptl4.logify ();
    ptl3.multiplyBy (ptl4);
    assertTrue (ptl3.almostEquals (ptl1));

    MultinomialPotential ptl5 = new MultinomialPotential (var, vals);
    MultinomialPotential ptl6 = new MultinomialPotential (var, vals2);
    ptl6.logify ();
    ptl6.multiplyBy (ptl5);
    ptl6.delogify ();
    assertTrue (ptl6.almostEquals (ans));

    MultinomialPotential ptl7 = new MultinomialPotential (var, vals);
    MultinomialPotential ptl8 = new MultinomialPotential (var, vals2);
    ptl7.logify (); ptl8.logify ();
    ptl8.multiplyBy (ptl7);
    ptl8.delogify ();
    assertTrue (ptl8.almostEquals (ans));
  }

  public void testDivideByLogSpace ()
  {
    Variable var = new Variable (4);
    double[] vals = new double[]{ 2.0, 4.0, 6.0, 8.0 };
    double[] vals2 = new double [] { 0.5, 0.5, 0.5, 0.5 };

    double[] vals3 = new double [] { 4, 8, 12, 16, };
    MultinomialPotential ans = new MultinomialPotential (var, vals3);

    MultinomialPotential ptl1 = new MultinomialPotential (var, vals);
    MultinomialPotential ptl2 = new MultinomialPotential (var, vals2);
    ptl1.divideBy (ptl2);
    assertTrue (ans.almostEquals (ptl1));

    MultinomialPotential ptl3 = new MultinomialPotential (var, vals);
    MultinomialPotential ptl4 = new MultinomialPotential (var, vals2);
    ptl4.logify ();
    ptl3.divideBy (ptl4);
    assertTrue (ptl3.almostEquals (ans));

    MultinomialPotential ptl5 = new MultinomialPotential (var, vals);
    MultinomialPotential ptl6 = new MultinomialPotential (var, vals2);
    ptl5.logify ();
    ptl5.divideBy (ptl6);
    ptl5.delogify ();
    assertTrue (ptl5.almostEquals (ans));

    MultinomialPotential ptl7 = new MultinomialPotential (var, vals);
    MultinomialPotential ptl8 = new MultinomialPotential (var, vals2);
    ptl7.logify (); ptl8.logify ();
    ptl7.divideBy (ptl8);
    ptl7.delogify ();
    assertTrue (ptl7.almostEquals (ans));
  }


  public static Test suite ()
  {
    return new TestSuite (TestMultinomialPotential.class);
  }

  public static void main (String[] args) throws Throwable
  {
    TestSuite theSuite;
    if (args.length > 0) {
      theSuite = new TestSuite ();
      for (int i = 0; i < args.length; i++) {
        theSuite.addTest (new TestMultinomialPotential (args[i]));
      }
    } else {
      theSuite = (TestSuite) suite ();
    }

    junit.textui.TestRunner.run (theSuite);
  }

}