Package org.grouplens.lenskit

Examples of org.grouplens.lenskit.ItemScorer

80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
        assertEquals(RATINGS_DAT_MEAN, pv.get(2l), 0.00001);
    }

    @Test
    public void testUserMeanBaseline() {
        ItemScorer mean = makeGlobalMean();
        ItemScorer pred = new UserMeanItemScorer(new PrefetchingUserEventDAO(dao),
                                                 mean, new RatingVectorUserHistorySummarizer(),
                                                 0);
        // unseen item
        assertThat(pred.score(8, 4), closeTo(4.5, 0.001));
        // seen item - should be same avg
        assertThat(pred.score(8, 10), closeTo(4.5, 0.001));
        // unseen user - should be global mean
        assertThat(pred.score(10, 10), closeTo(RATINGS_DAT_MEAN, 0.001));
    }
View Full Code Here
97
98
99
100
101
102
103
104
105
106
107
108
    /**
     * Test falling back to an empty user.
     */
    @Test
    public void testUserMeanBaselineFallback() {
        ItemScorer mean = makeGlobalMean();
        ItemScorer pred = new UserMeanItemScorer(new PrefetchingUserEventDAO(dao),
                                                 mean, new RatingVectorUserHistorySummarizer(),
                                                 0);
        // unseen user - should be global mean
        assertThat(pred.score(10, 10), closeTo(RATINGS_DAT_MEAN, 0.001));
    }
View Full Code Here
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
        assertThat(pred.score(10, 10), closeTo(RATINGS_DAT_MEAN, 0.001));
    }

    @Test
    public void testItemMeanBaseline() {
        ItemScorer pred = new ItemMeanRatingItemScorer.Builder(dao, 0.0).get();
        long[] items = {5, 7, 10};
        double[] values = {3, 6, 4};
        SparseVector map = MutableSparseVector.wrap(items, values).freeze();
        // unseen item, should be global mean
        assertThat(pred.score(10, 2),
                   closeTo(RATINGS_DAT_MEAN, 0.001));
        // seen item - should be item average
        assertThat(pred.score(10, 5),
                   closeTo(3.0, 0.001));
    }
View Full Code Here
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
                   closeTo(3.0, 0.001));
    }

    @Test
    public void testUserItemMeanBaseline() {
        ItemScorer base = new ItemMeanRatingItemScorer.Builder(dao, 0.0).get();
        ItemScorer pred = new UserMeanItemScorer(new PrefetchingUserEventDAO(dao),
                                                 base, new RatingVectorUserHistorySummarizer(),
                                                 0);
        // we use user 8 - their average offset is 0.5
        // unseen item, should be global mean + user offset
        assertThat(pred.score(8, 10),
                   closeTo(RATINGS_DAT_MEAN + 0.5, 0.001));

        // seen item - should be item average + user offset
        assertThat(pred.score(8, 5),
                   closeTo(3.5, 0.001));
    }
View Full Code Here
31
32
33
34
35
36
37
38
39
import static org.junit.Assert.assertThat;

public class PrecomputedItemScorerTest {
    @Test
    public void testEmptyScorer() {
        ItemScorer scorer = PrecomputedItemScorer.newBuilder().build();
        assertThat(scorer.score(42, 1),
                   notANumber());
    }
View Full Code Here
38
39
40
41
42
43
44
45
46
47
48
49
50
                   notANumber());
    }

    @Test
    public void testAddScore() {
        ItemScorer scorer = PrecomputedItemScorer.newBuilder()
                                          .addScore(42, 1, 4)
                                          .build();
        assertThat(scorer.score(42, 1),
                   closeTo(4, 1.0e-5));
        assertThat(scorer.score(42, 2), notANumber());
        assertThat(scorer.score(39, 1), notANumber());
    }
View Full Code Here
39
40
41
42
43
44
45
46
*/
public class ConstantItemScorerTest {

    @Test
    public void testSingleScore() {
        ItemScorer pred = new ConstantItemScorer(5);
        assertThat(pred.score(5, 10), equalTo(5.0));
    }
View Full Code Here
49
50
51
52
53
54
55
56
57
58
59
60
61
62
        assertThat(scorer.score(39, 1), notANumber());
    }

    @Test
    public void testVectorScore() {
        ItemScorer scorer = PrecomputedItemScorer.newBuilder()
                                          .addScore(42, 1, 4)
                                          .build();
        MutableSparseVector msv = MutableSparseVector.create(1, 3);
        scorer.score(42, msv);
        assertThat(msv.containsKey(1), equalTo(true));
        assertThat(msv.containsKey(3), equalTo(false));
        assertThat(msv.get(1), closeTo(4, 1.0e-5));
    }
View Full Code Here
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
        assertThat(msv.get(1), closeTo(4, 1.0e-5));
    }

    @Test
    public void testAddMultipleScores() {
        ItemScorer scorer = PrecomputedItemScorer.newBuilder()
                                          .addScore(42, 3, 4)
                                          .addScore(42, 7, 2)
                                          .build();
        MutableSparseVector msv = MutableSparseVector.create(1, 3, 5, 7, 8);
        scorer.score(42, msv);
        assertThat(msv.keySet().size(), equalTo(2));
        assertThat(msv.containsKey(1), equalTo(false));
        assertThat(msv.containsKey(3), equalTo(true));
        assertThat(msv.containsKey(5), equalTo(false));
        assertThat(msv.containsKey(7), equalTo(true));
        assertThat(msv.containsKey(8), equalTo(false));
        assertThat(msv.get(3), closeTo(4, 1.0e-5));
        assertThat(msv.get(7), closeTo(2, 1.0e-5));

        scorer.score(5, msv);
        assertThat(msv.keySet().isEmpty(), equalTo(true));
    }
View Full Code Here
45
46
47
48
49
50
51
52
53
54
        assertThat(pred.score(5, 10), equalTo(5.0));
    }

    @Test
    public void testScoreSet() {
        ItemScorer pred = new ConstantItemScorer(5);
        SparseVector v = pred.score(42, LongUtils.packedSet(1, 2, 3, 5, 7));
        assertThat(v.keySet(), contains(1L, 2L, 3L, 5L, 7L));
        assertThat(v.values(), everyItem(equalTo(5.0)));
    }
View Full Code Here
TOP

Related Classes of org.grouplens.lenskit.ItemScorer

  • org.grouplens.lenskit.baseline.ConstantItemScorerTest
  • org.grouplens.lenskit.baseline.MeanScorersTest
  • org.grouplens.lenskit.basic.PrecomputedItemScorerTest
  • org.grouplens.lenskit.basic.SimpleRatingPredictorTest
  • org.grouplens.lenskit.mf.funksvd.FunkSVDRecommenderBuildTest
  • org.grouplens.lenskit.predict.ordrec.OrdRecRatingPredictorTest
  • org.grouplens.lenskit.slopeone.SlopeOneItemScorerTest
  • org.grouplens.lenskit.slopeone.WeightedSlopeOneItemScorerTest
    • 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.