Package tv.floe.metronome.linearregression

Source Code of tv.floe.metronome.linearregression.TestHypothesisFunction

package tv.floe.metronome.linearregression;

import static org.junit.Assert.*;

import org.apache.mahout.math.DenseMatrix;
import org.apache.mahout.math.DenseVector;
import org.apache.mahout.math.Matrix;
import org.apache.mahout.math.Vector;
import org.junit.Test;

/**
* In the case of linear regression this is simply the dot product of the
* current instance and the current parameter vector
*
*
*
* @author josh
*
*/
public class TestHypothesisFunction {

  @Test
  public void testMahoutLinearCombinaton() {
   
    System.out.println( "> testMahoutLinearCombinaton" );
   
    Matrix m = new DenseMatrix( 1, 2 );
    m.set(0, 0, 0.2);
    m.set(0, 1, 0.3);
   
    Vector v = new DenseVector(2);
    v.set(0, 0.1);
    v.set(1, 0.1);
   
    // returns the dot product for each row in the matrix as a vector
    Vector v2 = m.times(v);
   
    System.out.println("> " + m.get(0, 0) );
    System.out.println("> " + m.get(0, 1) );
   
    System.out.println("> Vector out ------ " );
    System.out.println("> size: " + v2.size() );
    System.out.println("> c0: " + v2.get(0) );

    assertEquals( v2.get(0), 0.05, 0.0 );
   
  }
 
}
TOP

Related Classes of tv.floe.metronome.linearregression.TestHypothesisFunction

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.