@Override public boolean hasNext() { return longIterator.hasNext(); }
private void saveDrfIds() { for (LongIterator i = this.drfIds.iterator(); i.hasNext();) { writeIFRecord(IFREC_DRF_CREATE, i.next()); this.ifLiveRecordCount++; this.ifTotalRecordCount++; } }
private void saveCrfIds() { for (LongIterator i = this.crfIds.iterator(); i.hasNext();) { writeIFRecord(IFREC_CRF_CREATE, i.next()); this.ifLiveRecordCount++; this.ifTotalRecordCount++; } }
@Override public DateTimeColumn removeMissing() { DateTimeColumn noMissing = emptyCopy(); LongIterator iterator = longIterator(); while(iterator.hasNext()) { long i = iterator.nextLong(); if (!valueIsMissing(i)) { noMissing.appendInternal(i); } } return noMissing; }
@Override public boolean hasNext() { return iter.hasNext(); }
@Override public boolean hasNext() { return idIter.hasNext(); }
@Override public boolean hasNext() { return iter.hasNext(); }
@Nonnull @Override public ResultMap scoreWithDetails(long user, @Nonnull Collection<Long> items) { List<Result> results = new ArrayList<>(items.size()); LongIterator iter = LongIterators.asLongIterator(items.iterator()); while (iter.hasNext()) { long item = iter.nextLong(); results.add(Results.create(item, fixedScore)); } return Results.newResultMap(results); }
@Nonnull @Override public ResultMap scoreWithDetails(long user, @Nonnull Collection<Long> items) { List<Result> results = new ArrayList<>(items.size()); LongIterator iter = LongIterators.asLongIterator(items.iterator()); while (iter.hasNext()) { long item = iter.nextLong(); results.add(Results.create(item, rankScores.get(item))); } return Results.newResultMap(results); } }
@Override public String get() { String name = fileName; logger.info("writing {}", name); File file = new File(workingDir, name); try (PrintWriter writer = new PrintWriter(file, CHARSET_UTF_8)) { LongIterator iter = identifiers.iterator(); while (iter.hasNext()) { writer.println(iter.nextLong()); } } catch (IOException e) { throw new ExternalProcessException("Error creating ratings file", e); } return name; } }
@Nonnull @Override public ResultMap scoreWithDetails(long user, @Nonnull Collection<Long> items) { List<Result> rs = new ArrayList<>(items.size()); KeyedObjectMap<Result> userResults = userData.get(user); if (userResults != null) { LongIterator iter = LongIterators.asLongIterator(items.iterator()); while (iter.hasNext()) { long item = iter.nextLong(); Result r = userResults.get(item); if (r != null) { rs.add(r); } } } return Results.newResultMap(rs); }
@Nonnull @Override public ResultMap scoreWithDetails(long user, @Nonnull Collection<Long> items) { double userScore = globalMean + userBiases.get(user); List<Result> results = new ArrayList<>(); LongIterator iter = LongIterators.asLongIterator(items.iterator()); while (iter.hasNext()) { long item = iter.nextLong(); double score = userScore + itemBiases.get(item); results.add(Results.create(item, score)); } return Results.newResultMap(results); }
@Nonnull @Override public ResultMap scoreWithDetails(long user, @Nonnull Collection<Long> items) { List<Result> results = new ArrayList<>(); double base = model.getIntercept() + model.getUserBias(user); LongIterator iter = LongIterators.asLongIterator(items.iterator()); while (iter.hasNext()) { long item = iter.nextLong(); results.add(Results.create(item, base + model.getItemBias(item))); } return Results.newResultMap(results); } }
/** * Score items into an accumulator. * @param basket The basket of reference items. * @param items The item scores. * @param accum The accumulator. */ private void scoreItems(@Nonnull Collection<Long> basket, Collection<Long> items, ItemItemScoreAccumulator accum) { LongSet bset = LongUtils.packedSet(basket); Long2DoubleMap basketScores = LongUtils.constantDoubleMap(bset, 1.0); LongIterator iter = LongIterators.asLongIterator(items.iterator()); while (iter.hasNext()) { long item = iter.nextLong(); scoreItem(basketScores, item, accum); } }
@Nonnull @Override public ResultMap scoreWithDetails(long user, @Nonnull Collection<Long> items) { final double gmean = summary.getGlobalMean(); List<Result> results = new ArrayList<>(); LongIterator iter = LongIterators.asLongIterator(items.iterator()); while (iter.hasNext()) { final long item = iter.nextLong(); double offset = summary.getItemOffset(item); if (!Scalars.isZero(damping)) { int count = summary.getItemRatingCount(item); offset = offset * count / (count + damping); } results.add(Results.create(item, gmean + offset)); } return Results.newResultMap(results); }
/** * Construct a new indexer. It maps each long key to its position in the list, and then be ready for more keys. * * @param keys The list of keys to store. */ public static HashKeyIndex create(LongList keys) { HashKeyIndex idx = create(); LongIterator iter = keys.iterator(); while (iter.hasNext()) { idx.internId(iter.nextLong()); } return idx; }
@Override protected Neighbor computeNext() { while (neighborIter.hasNext()) { final long neighbor = neighborIter.nextLong(); Long2DoubleMap vector = snapshot.getNormalizedUserVector(neighbor); double sim = similarity.similarity(user, userVector, neighbor, vector); if (acceptSimilarity(sim)) { return new Neighbor(neighbor, snapshot.getUserVector(neighbor), sim); } } // no neighbor found, done return endOfData(); } }
@Test public void testEmptyUserSet() { LongIterator iter = new AdaptiveSparseItemIterator(context, LongSets.EMPTY_SET); assertThat(iter.hasNext(), equalTo(false)); try { iter.nextLong(); fail("nextLong() should throw NSEE"); } catch (NoSuchElementException e) { /* expected */ } }
private Long2DoubleSortedArrayMap slowSubMap(LongSet toKeep) { LongSortedSet kept = LongUtils.setIntersect(keySet(), toKeep); double[] nvs = new double[kept.size()]; int i = keys.getLowerBound(); int j = 0; LongIterator iter = kept.iterator(); while (iter.hasNext()) { long key = iter.nextLong(); while (keys.getKey(i) < key) { i++; } nvs[j] = values[i]; j++; i++; } return wrap(SortedKeyIndex.fromCollection(kept), nvs); }