Package org.grouplens.lenskit

Examples of org.grouplens.lenskit.ItemScorer.score()

74
75
76
77
78
79
80
81
82
83
84
    }

    @Test
    public void testMeanBaseline() {
        ItemScorer pred = makeGlobalMean();
        SparseVector pv = pred.score(10L, itemSet(2l));
        assertEquals(RATINGS_DAT_MEAN, pv.get(2l), 0.00001);
    }

    @Test
    public void testUserMeanBaseline() {
View Full Code Here
85
86
87
88
89
90
91
92
93
94
95
        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
87
88
89
90
91
92
93
94
95
96
97
                                                 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
89
90
91
92
93
94
95
96
97
98
99
        // 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));
    }

    /**
     * Test falling back to an empty user.
     */
 
View Full Code Here
102
103
104
105
106
107
108
109
110
111
112
        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));
    }

    @Test
    public void testItemMeanBaseline() {
        ItemScorer pred = new ItemMeanRatingItemScorer.Builder(dao, 0.0).get();
View Full Code Here
112
113
114
115
116
117
118
119
120
121
122
        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
115
116
117
118
119
120
121
122
123
124
125
        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));
    }

    @Test
    public void testUserItemMeanBaseline() {
View Full Code Here
127
128
129
130
131
132
133
134
135
136
137
        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
131
132
133
134
135
136
137
138
139
        // 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
32
33
34
35
36
37
38
39
40
41
42
public class PrecomputedItemScorerTest {
    @Test
    public void testEmptyScorer() {
        ItemScorer scorer = PrecomputedItemScorer.newBuilder().build();
        assertThat(scorer.score(42, 1),
                   notANumber());
    }

    @Test
    public void testAddScore() {
View Full Code Here
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.