Package org.apache.mahout.cf.taste.model

Examples of org.apache.mahout.cf.taste.model.User


    // Yeah, undefined in this case
    assertTrue(Double.isNaN(correlation));
  }

  public void testNoItemCorrelation1() throws Exception {
    User user1 = getUser("test1", 3.0, -3.0);
    User user2 = getUser("test2", -2.0, 2.0);
    DataModel dataModel = getDataModel(user1, user2);
    double correlation =
            new PearsonCorrelationSimilarity(dataModel).itemSimilarity(dataModel.getItem("0"), dataModel.getItem("1"));
    assertCorrelationEquals(-1.0, correlation);
  }
View Full Code Here


            new PearsonCorrelationSimilarity(dataModel).itemSimilarity(dataModel.getItem("1"), dataModel.getItem("2"));
    assertTrue(Double.isNaN(correlation));
  }

  public void testNoItemCorrelation3() throws Exception {
    User user1 = getUser("test1", 90.0, 70.0);
    User user2 = getUser("test2", 80.0, 80.0);
    User user3 = getUser("test3", 70.0, 90.0);
    DataModel dataModel = getDataModel(user1, user2, user3);
    double correlation =
            new PearsonCorrelationSimilarity(dataModel).itemSimilarity(dataModel.getItem("0"), dataModel.getItem("1"));
    assertCorrelationEquals(-1.0, correlation);
  }
View Full Code Here

            new PearsonCorrelationSimilarity(dataModel).itemSimilarity(dataModel.getItem("0"), dataModel.getItem("1"));
    assertCorrelationEquals(-1.0, correlation);
  }

  public void testSimpleItem() throws Exception {
    User user1 = getUser("test1", 1.0, 2.0);
    User user2 = getUser("test2", 2.0, 5.0);
    User user3 = getUser("test3", 3.0, 6.0);
    DataModel dataModel = getDataModel(user1, user2, user3);
    double correlation =
            new PearsonCorrelationSimilarity(dataModel).itemSimilarity(dataModel.getItem("0"), dataModel.getItem("1"));
    assertCorrelationEquals(0.9607689228305227, correlation);
  }
View Full Code Here

            new PearsonCorrelationSimilarity(dataModel).itemSimilarity(dataModel.getItem("0"), dataModel.getItem("1"));
    assertCorrelationEquals(0.9607689228305227, correlation);
  }

  public void testSimpleItemWeighted() throws Exception {
    User user1 = getUser("test1", 1.0, 2.0);
    User user2 = getUser("test2", 2.0, 5.0);
    User user3 = getUser("test3", 3.0, 6.0);
    DataModel dataModel = getDataModel(user1, user2, user3);
    ItemSimilarity itemSimilarity = new PearsonCorrelationSimilarity(dataModel, Weighting.WEIGHTED);
    double correlation = itemSimilarity.itemSimilarity(dataModel.getItem("0"), dataModel.getItem("1"));
    assertCorrelationEquals(0.9901922307076306, correlation);
  }
View Full Code Here

* <p>Tests {@link LogLikelihoodSimilarity}.</p>
*/
public final class LogLikelihoodSimilarityTest extends SimilarityTestCase {

  public void testCorrelation() throws Exception {
    User user1 = getUser("test1", 1.01.0);
    User user2 = getUser("test2", 1.0,  null, 1.0);
    User user3 = getUser("test3", null, null, 1.01.01.0);
    User user4 = getUser("test4", 1.01.01.01.01.0);
    User user5 = getUser("test5", null, 1.01.01.01.0);
    DataModel dataModel = getDataModel(user1, user2, user3, user4, user5);

    double correlation = new LogLikelihoodSimilarity(dataModel).
        itemSimilarity(dataModel.getItem("1"), dataModel.getItem("0"));
    assertCorrelationEquals(0.12160727029227925, correlation);
View Full Code Here

* <p>Tests {@link EuclideanDistanceSimilarity}.</p>
*/
public final class EuclideanDistanceSimilarityTest extends SimilarityTestCase {

  public void testFullCorrelation1() throws Exception {
    User user1 = getUser("test1", 3.0, -2.0);
    User user2 = getUser("test2", 3.0, -2.0);
    DataModel dataModel = getDataModel(user1, user2);
    double correlation = new EuclideanDistanceSimilarity(dataModel).userSimilarity(user1, user2);
    assertCorrelationEquals(1.0, correlation);
  }
View Full Code Here

    double correlation = new EuclideanDistanceSimilarity(dataModel).userSimilarity(user1, user2);
    assertCorrelationEquals(1.0, correlation);
  }

  public void testFullCorrelation1Weighted() throws Exception {
    User user1 = getUser("test1", 3.0, -2.0);
    User user2 = getUser("test2", 3.0, -2.0);
    DataModel dataModel = getDataModel(user1, user2);
    double correlation = new EuclideanDistanceSimilarity(dataModel, Weighting.WEIGHTED).userSimilarity(user1, user2);
    assertCorrelationEquals(1.0, correlation);
  }
View Full Code Here

    double correlation = new EuclideanDistanceSimilarity(dataModel, Weighting.WEIGHTED).userSimilarity(user1, user2);
    assertCorrelationEquals(1.0, correlation);
  }

  public void testFullCorrelation2() throws Exception {
    User user1 = getUser("test1", 3.0, 3.0);
    User user2 = getUser("test2", 3.0, 3.0);
    DataModel dataModel = getDataModel(user1, user2);
    double correlation = new EuclideanDistanceSimilarity(dataModel).userSimilarity(user1, user2);
    assertTrue(Double.isNaN(correlation));
  }
View Full Code Here

    double correlation = new EuclideanDistanceSimilarity(dataModel).userSimilarity(user1, user2);
    assertTrue(Double.isNaN(correlation));
  }

  public void testNoCorrelation1() throws Exception {
    User user1 = getUser("test1", 3.0, -2.0);
    User user2 = getUser("test2", -3.0, 2.0);
    DataModel dataModel = getDataModel(user1, user2);
    double correlation = new EuclideanDistanceSimilarity(dataModel).userSimilarity(user1, user2);
    assertCorrelationEquals(0.424465381883345, correlation);
  }
View Full Code Here

    double correlation = new EuclideanDistanceSimilarity(dataModel).userSimilarity(user1, user2);
    assertCorrelationEquals(0.424465381883345, correlation);
  }

  public void testNoCorrelation1Weighted() throws Exception {
    User user1 = getUser("test1", 3.0, -2.0);
    User user2 = getUser("test2", -3.0, 2.0);
    DataModel dataModel = getDataModel(user1, user2);
    double correlation = new EuclideanDistanceSimilarity(dataModel, Weighting.WEIGHTED).userSimilarity(user1, user2);
    assertCorrelationEquals(0.8081551272944483, correlation);
  }
View Full Code Here

TOP

Related Classes of org.apache.mahout.cf.taste.model.User

Copyright © 2018 www.massapicom. 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.