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

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

123
124
125
126
127
128
129
130
131
132
133
    double correlation = new PearsonCorrelationSimilarity(dataModel, Weighting.WEIGHTED).userSimilarity(1, 2);
    assertCorrelationEquals(0.9901922307076306, correlation);
  }

  public void testFullItemCorrelation1() throws Exception {
    DataModel dataModel = getDataModel(
            new long[] {1, 2},
            new Double[][] {
                    {3.0, 3.0},
                    {-2.0, -2.0},
            });
View Full Code Here
135
136
137
138
139
140
141
142
143
144
145
        new PearsonCorrelationSimilarity(dataModel).itemSimilarity(0, 1);
    assertCorrelationEquals(1.0, correlation);
  }

  public void testFullItemCorrelation2() throws Exception {
    DataModel dataModel = getDataModel(
            new long[] {1, 2},
            new Double[][] {
                    {3.0, 3.0},
                    {3.0, 3.0},
            });
View Full Code Here
148
149
150
151
152
153
154
155
156
157
158
    // Yeah, undefined in this case
    assertTrue(Double.isNaN(correlation));
  }

  public void testNoItemCorrelation1() throws Exception {
    DataModel dataModel = getDataModel(
            new long[] {1, 2},
            new Double[][] {
                    {3.0, -3.0},
                    {2.0, -2.0},
            });
View Full Code Here
160
161
162
163
164
165
166
167
168
169
170
        new PearsonCorrelationSimilarity(dataModel).itemSimilarity(0, 1);
    assertCorrelationEquals(-1.0, correlation);
  }

  public void testNoItemCorrelation2() throws Exception {
    DataModel dataModel = getDataModel(
            new long[] {1, 2},
            new Double[][] {
                    {null, 1.0, null},
                    {null, null, 1.0},
            });
View Full Code Here
172
173
174
175
176
177
178
179
180
181
182
        new PearsonCorrelationSimilarity(dataModel).itemSimilarity(1, 2);
    assertTrue(Double.isNaN(correlation));
  }

  public void testNoItemCorrelation3() throws Exception {
    DataModel dataModel = getDataModel(
            new long[] {1, 2, 3},
            new Double[][] {
                    {90.0, 70.0},
                    {80.0, 80.0},
                    {70.0, 90.0},
View Full Code Here
185
186
187
188
189
190
191
192
193
194
195
        new PearsonCorrelationSimilarity(dataModel).itemSimilarity(0, 1);
    assertCorrelationEquals(-1.0, correlation);
  }

  public void testSimpleItem() throws Exception {
    DataModel dataModel = getDataModel(
            new long[] {1, 2, 3},
            new Double[][] {
                    {1.0, 2.0},
                    {2.0, 5.0},
                    {3.0, 6.0},
View Full Code Here
198
199
200
201
202
203
204
205
206
207
208
        new PearsonCorrelationSimilarity(dataModel).itemSimilarity(0, 1);
    assertCorrelationEquals(0.9607689228305227, correlation);
  }

  public void testSimpleItemWeighted() throws Exception {
    DataModel dataModel = getDataModel(
            new long[] {1, 2, 3},
            new Double[][] {
                    {1.0, 2.0},
                    {2.0, 5.0},
                    {3.0, 6.0},
View Full Code Here
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
/** <p>Tests {@link InverseUserFrequency}.</p> */
public final class InverseUserFrequencyTest extends TransformTestCase {

  public void testIUF() throws Exception {
    DataModel dataModel = getDataModel(
            new long[] {1, 2, 3, 4, 5},
            new Double[][] {
                    {0.1},
                    {0.2, 0.3},
                    {0.4, 0.5, 0.6},
                    {0.7, 0.8, 0.9, 1.0},
                    {1.0, 1.0, 1.0, 1.0, 1.0},
            });

    InverseUserFrequency iuf = new InverseUserFrequency(dataModel, 10.0);

    PreferenceArray user5Prefs = dataModel.getPreferencesFromUser(5);

    for (int i = 0; i < 5; i++) {
      Preference pref = user5Prefs.get(i);
      assertNotNull(pref);
      assertEquals(Math.log(5.0 / (double) (5 - i)) / Math.log(iuf.getLogBase()),
View Full Code Here
21
22
23
24
25
26
27
28
29
30
31
/** <p>Tests {@link LogLikelihoodSimilarity}.</p> */
public final class LogLikelihoodSimilarityTest extends SimilarityTestCase {

  public void testCorrelation() throws Exception {
    DataModel dataModel = getDataModel(
            new long[] {1, 2, 3, 4, 5},
            new Double[][] {
                    {1.0, 1.0},
                    {1.0, null, 1.0},
                    {null, null, 1.0, 1.0, 1.0},
View Full Code Here
24
25
26
27
28
29
30
31
32
33
34
/** <p>Tests {@link EuclideanDistanceSimilarity}.</p> */
public final class EuclideanDistanceSimilarityTest extends SimilarityTestCase {

  public void testFullCorrelation1() throws Exception {
    DataModel dataModel = getDataModel(
            new long[] {1, 2},
            new Double[][] {
                    {3.0, -2.0},
                    {3.0, -2.0},
            });
View Full Code Here
TOP

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

  • com.manning.hip.ch9.MovieUserEvaluator
  • com.manning.hip.ch9.MovieUserRecommender
  • io.prediction.examples.java.recommendations.tutorial5.MahoutAlgorithm
  • mia.recommender.ch02.EvaluatorIntro
  • mia.recommender.ch02.IREvaluatorIntro
  • mia.recommender.ch02.ItemBaseRecommender
  • mia.recommender.ch02.ItemBaseRecommenderEvaluation
  • mia.recommender.ch02.UserBaseRecommender
  • mia.recommender.ch02.UserBaseRecommenderEvaluation
  • mia.recommender.ch03.CreateGenericDataModel
    • 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.