Package it.unimi.dsi.fastutil.longs

Examples of it.unimi.dsi.fastutil.longs.LongSortedSet

181
182
183
184
185
186
187
188
189
190
191
192
        return channels.keySet();
    }

    @Override
    public ImmutableSparseVector combineWith(SparseVector o) {
        LongSortedSet key = this.keyDomain();
        LongSortedSet newKey = o.keyDomain();
        MutableSparseVector result = MutableSparseVector.create(LongUtils.setUnion(key, newKey));
        result.set(this);
        result.set(o);
        return result.freeze();
    }
View Full Code Here
236
237
238
239
240
241
242
243
244
245
246
247
248
249
    @Test
    public void testMaskEmpty() {
        LongKeyDomain keys = LongKeyDomain.create(2, 7, 8, 42, 639);
        keys.setAllActive(false);
        LongSortedSet set = new LongSortedArraySet(keys);

        assertThat(set.size(), equalTo(0));
        assertTrue(set.isEmpty());
        assertFalse(set.iterator().hasNext());
        for (int i = 0; i < keys.domainSize(); i++) {
            assertFalse(set.contains(keys.getKey(i)));
        }
    }
View Full Code Here
252
253
254
255
256
257
258
259
260
261
262
    public void testMaskedIterator() {
        LongKeyDomain keys = LongKeyDomain.create(2, 7, 8, 42, 639);
        keys.setAllActive(true);
        keys.setActive(0, false);
        keys.setActive(4, false);
        LongSortedSet set = new LongSortedArraySet(keys);
        assertTrue(set.iterator(7).hasNext());
        assertTrue(set.iterator(7).hasPrevious());
        assertThat(set.iterator(7).nextLong(), equalTo(8l));
        assertThat(set.iterator(7).previousLong(), equalTo(7l));
    }
View Full Code Here
265
266
267
268
269
270
271
272
273
274
275
    public void testMaskedIteratorMid() {
        LongKeyDomain keys = LongKeyDomain.create(2, 7, 8, 42, 639);
        keys.setAllActive(true);
        keys.setActive(0, false);
        keys.setActive(2, false);
        LongSortedSet set = new LongSortedArraySet(keys);
        assertTrue(set.iterator(8).hasNext());
        assertTrue(set.iterator(8).hasPrevious());
        assertThat(set.iterator(8).nextLong(), equalTo(42l));
        assertThat(set.iterator(8).previousLong(), equalTo(7l));
    }
View Full Code Here
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
    /**
     * Write a user or item index to the file.
     * @param map The index to write.
     */
    private void writeIndex(Long2ObjectMap<IntList> map) throws IOException {
        LongSortedSet keys = LongUtils.packedSet(map.keySet());
        BinaryIndexTableWriter tableWriter =
                BinaryIndexTableWriter.create(format, channel, keys.size());

        SortComparator indexComparator = new SortComparator();

        LongIterator iter = keys.iterator();
        while (iter.hasNext()) {
            final long key = iter.nextLong();
            int[] indexes = map.get(key).toIntArray();
            if (needsSorting) {
                IntArrays.quickSort(indexes, indexComparator);
View Full Code Here
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
     * is a subset of the cached items, use the cache; otherwise score the
     * new items and update the cached scores.
     */
    @Override
    public void score(long user, @Nonnull MutableSparseVector scores){
        LongSortedSet reqItems = scores.keyDomain();
        if(cachedId == user && cachedScores != null) {
            LongSortedSet cachedItems = cachedScores.keyDomain();
            if(!cachedItems.containsAll(reqItems)) {
                LongSortedSet diffItems = LongUtils.setDifference(reqItems, cachedItems);
                SparseVector newCache = scorer.score(user, diffItems);
                cachedScores = cachedScores.combineWith(newCache);
            }
            scores.set(cachedScores);
        } else {
View Full Code Here
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
        SparseVector predictions = user.getPredictions();
        if (predictions == null) {
            predictions = MutableSparseVector.create();
        }

        LongSortedSet items = ratings.keySet();
        if (!items.containsAll(predictions.keySet())) {
            items = LongUtils.setUnion(items, predictions.keySet());
        }

        logger.debug("outputting {} predictions for user {}", predictions.size(), user.getUserId());
        for (LongIterator iter = items.iterator(); iter.hasNext(); /* no increment */) {
            long item = iter.nextLong();
            Double rating = ratings.containsKey(item) ? ratings.get(item) : null;
            Double pred = predictions.containsKey(item) ? predictions.get(item) : null;
            try {
                List<Object> row = Lists.<Object>newArrayList(user.getUserId(), item, rating, pred);
View Full Code Here
TOP

Related Classes of it.unimi.dsi.fastutil.longs.LongSortedSet

  • org.grouplens.lenskit.basic.SimpleCachingItemScorer
  • org.grouplens.lenskit.collections.LongSortedArraySetTest
  • org.grouplens.lenskit.collections.LongUtilsTest
  • org.grouplens.lenskit.data.dao.packed.BinaryRatingPacker
  • org.grouplens.lenskit.eval.traintest.OutputPredictMetric
  • org.grouplens.lenskit.knn.item.ItemItemGlobalScorer
  • org.grouplens.lenskit.knn.item.model.ItemItemBuildContext
  • org.grouplens.lenskit.knn.item.model.NormalizingItemItemModelBuilder
  • org.grouplens.lenskit.vectors.ImmutableSparseVector
  • org.grouplens.lenskit.vectors.SparseVectorTestCommon
    • 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.