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

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


            new EuclideanDistanceSimilarity(dataModel).itemSimilarity(dataModel.getItem("0"), dataModel.getItem("1"));
    assertCorrelationEquals(0.3606507916004517, 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 EuclideanDistanceSimilarity(dataModel).itemSimilarity(dataModel.getItem("0"), dataModel.getItem("1"));
    assertCorrelationEquals(0.5896248568217328, correlation);
  }
View Full Code Here


            new EuclideanDistanceSimilarity(dataModel).itemSimilarity(dataModel.getItem("0"), dataModel.getItem("1"));
    assertCorrelationEquals(0.5896248568217328, 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 EuclideanDistanceSimilarity(dataModel, Weighting.WEIGHTED);
    double correlation = itemSimilarity.itemSimilarity(dataModel.getItem("0"), dataModel.getItem("1"));
    assertCorrelationEquals(0.8974062142054332, correlation);
  }
View Full Code Here

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

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

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

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

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

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

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

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

    assertEquals(0.3, itemCorrelation.itemSimilarity(item1, item3));
    assertTrue(Double.isNaN(itemCorrelation.itemSimilarity(item3, item4)));
  }

  public void testFromCorrelation() 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 otherSimilarity = new PearsonCorrelationSimilarity(dataModel);
    ItemSimilarity itemSimilarity = new GenericItemSimilarity(otherSimilarity, dataModel);
    assertCorrelationEquals(1.0,
                            itemSimilarity.itemSimilarity(dataModel.getItem("0"), dataModel.getItem("0")));
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.