Package org.grouplens.lenskit.vectors

Examples of org.grouplens.lenskit.vectors.MutableSparseVector

118
119
120
121
122
123
124
125
126
127
128
129
130
        List<Rating> ratings = new ArrayList<Rating>();
        ratings.add(new SimpleRating(1, 2, 3.0, 3));
        ratings.add(new SimpleRating(1, 5, 2.3, 4));
        ratings.add(new SimpleRating(1, 3, 4.5, 5));
        ratings.add(new SimpleRating(1, 5, 3.7, 6));
        MutableSparseVector urv = Ratings.userRatingVector(ratings);
        assertThat(urv.isEmpty(), equalTo(false));
        assertThat(urv.size(), equalTo(3));
        assertThat(urv.get(2), closeTo(3.0, 1.0e-6));
        assertThat(urv.get(3), closeTo(4.5, 1.0e-6));
        assertThat(urv.get(5), closeTo(3.7, 1.0e-6));
        assertThat(urv.containsKey(1), equalTo(false));
    }
View Full Code Here
134
135
136
137
138
139
140
141
142
143
144
145
146
        List<Rating> ratings = new ArrayList<Rating>();
        ratings.add(new SimpleRating(1, 2, 3.0, 3));
        ratings.add(new SimpleRating(1, 5, 2.3, 5));
        ratings.add(new SimpleNullRating(1, 2, 7));
        ratings.add(new SimpleRating(1, 3, 4.5, 8));
        MutableSparseVector urv = Ratings.userRatingVector(ratings);
        assertThat(urv.isEmpty(), equalTo(false));
        assertThat(urv.size(), equalTo(2));
        assertThat(urv.get(3), closeTo(4.5, 1.0e-6));
        assertThat(urv.get(5), closeTo(2.3, 1.0e-6));
        assertThat(urv.containsKey(1), equalTo(false));
        assertThat(urv.containsKey(2), equalTo(false));
    }
View Full Code Here
52
53
54
55
56
57
58
59
60
61
62
    @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
150
151
152
153
154
155
156
157
158
159
160
161
162
        List<Rating> ratings = new ArrayList<Rating>();
        ratings.add(new SimpleRating(1, 2, 3.0, 3));
        ratings.add(new SimpleRating(1, 5, 2.3, 7));
        ratings.add(new SimpleRating(1, 3, 4.5, 5));
        ratings.add(new SimpleRating(1, 5, 3.7, 6));
        MutableSparseVector urv = Ratings.userRatingVector(ratings);
        assertThat(urv.isEmpty(), equalTo(false));
        assertThat(urv.size(), equalTo(3));
        assertThat(urv.get(2), closeTo(3.0, 1.0e-6));
        assertThat(urv.get(3), closeTo(4.5, 1.0e-6));
        assertThat(urv.get(5), closeTo(2.3, 1.0e-6));
        assertThat(urv.containsKey(1), equalTo(false));
    }
View Full Code Here
54
55
56
57
58
59
60
61
62
63
    }

    @Test
    public void testScoreVector() {
        ItemScorer pred = new ConstantItemScorer(5);
        MutableSparseVector v = MutableSparseVector.create(1, 2, 3, 5, 7);
        pred.score(42, v);
        assertThat(v.keySet(), contains(1L, 2L, 3L, 5L, 7L));
        assertThat(v.values(), everyItem(equalTo(5.0)));
    }
View Full Code Here
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
    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
162
163
164
165
166
167
168
169
170
    }

    @Test
    public void testEmptyIRV() {
        List<Rating> ratings = Collections.emptyList();
        MutableSparseVector urv = Ratings.itemRatingVector(ratings);
        assertThat(urv.isEmpty(), equalTo(true));
        assertThat(urv.size(), equalTo(0));
    }
View Full Code Here
174
175
176
177
178
179
180
181
182
183
184
185
186
        List<Rating> ratings = new ArrayList<Rating>();
        ratings.add(new SimpleRating(1, 2, 3.0, 1));
        ratings.add(new SimpleRating(3, 2, 4.5, 2));
        ratings.add(new SimpleRating(2, 2, 2.3, 3));
        ratings.add(new SimpleRating(3, 2, 4.5, 10));
        MutableSparseVector urv = Ratings.itemRatingVector(ratings);
        assertThat(urv.isEmpty(), equalTo(false));
        assertThat(urv.size(), equalTo(3));
        assertThat(urv.get(1), closeTo(3.0, 1.0e-6));
        assertThat(urv.get(3), closeTo(4.5, 1.0e-6));
        assertThat(urv.get(2), closeTo(2.3, 1.0e-6));
        assertThat(urv.containsKey(5), equalTo(false));
    }
View Full Code Here
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
        assertThat(msv.keySet().isEmpty(), equalTo(true));
    }

    @Test
    public void testAddScoreVector() {
        MutableSparseVector uv = MutableSparseVector.create(2, 3, 5);
        uv.set(2, 3);
        uv.set(3, 4);
        uv.set(5, 2);
        ItemScorer scorer = PrecomputedItemScorer.newBuilder()
                                          .addUser(3, uv)
                                          .build();
        MutableSparseVector output = MutableSparseVector.create(2, 3, 4, 5);
        scorer.score(3, output);
        assertThat(output, equalTo(uv));
    }
View Full Code Here
79
80
81
82
83
84
85
86
87
88
89
90
91
    }

    @Test
    public void testKnownUserItem() {
        final long user = 1;
        MutableSparseVector output = MutableSparseVector.create(1, 2, 3);

        predictor.score(user, output);

        assertEquals(3.18, output.get(1), EPSILON);
        assertEquals(3.04, output.get(2), EPSILON);
        assertEquals(3.07, output.get(3), EPSILON);
    }
View Full Code Here
TOP

Related Classes of org.grouplens.lenskit.vectors.MutableSparseVector

  • org.grouplens.lenskit.baseline.ConstantItemScorerTest
  • org.grouplens.lenskit.baseline.FallbackItemScorer
  • org.grouplens.lenskit.baseline.FallbackItemScorerTest
  • org.grouplens.lenskit.baseline.LeastSquaresItemScorer$Builder
  • org.grouplens.lenskit.baseline.LeastSquaresItemScorerTest
  • org.grouplens.lenskit.baseline.UserMeanItemScorer
  • org.grouplens.lenskit.basic.AbstractGlobalItemScorer
  • org.grouplens.lenskit.basic.AbstractItemScorer
  • org.grouplens.lenskit.basic.AbstractRatingPredictor
  • org.grouplens.lenskit.basic.PrecomputedItemScorerTest
    • 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.