Package org.apache.mahout.cf.taste.impl.common

Examples of org.apache.mahout.cf.taste.impl.common.FastIDSet.addAll()

36
37
38
39
40
41
42
43
44
45
46
public final class AllUnknownItemsCandidateItemsStrategyTest extends TasteTestCase {

  @Test 
  public void testStrategy() throws TasteException {
    FastIDSet allItemIDs = new FastIDSet();
    allItemIDs.addAll(new long[] { 1L, 2L, 3L });

    FastIDSet preferredItemIDs = new FastIDSet(1);
    preferredItemIDs.add(2L);
   
    DataModel dataModel = EasyMock.createMock(DataModel.class);
View Full Code Here
64
65
66
67
68
69
70
71
72
73
74
  protected FastIDSet getAllOtherItems(long[] theNeighborhood, long theUserID, boolean includeKnownItems)
      throws TasteException {
    DataModel dataModel = getDataModel();
    FastIDSet possibleItemIDs = new FastIDSet();
    for (long userID : theNeighborhood) {
      possibleItemIDs.addAll(dataModel.getItemIDsFromUser(userID));
    }
    if (!includeKnownItems) {
      possibleItemIDs.removeAll(dataModel.getItemIDsFromUser(theUserID));
    }
    return possibleItemIDs;
View Full Code Here
192
193
194
195
196
197
198
199
200
201
202
  private FastIDSet extractAllItemIdsFromUserItemFilter(Map<Long, FastIDSet> filter) {
    FastIDSet result = new FastIDSet();

    for (FastIDSet itemIds : filter.values()) {
      result.addAll(itemIds);
    }

    return result;
  }
View Full Code Here
169
170
171
172
173
174
175
176
177
178
179
  protected FastIDSet getAllOtherItems(long[] theNeighborhood, long theUserID, boolean includeKnownItems)
    throws TasteException {
    DataModel dataModel = getDataModel();
    FastIDSet possibleItemIDs = new FastIDSet();
    for (long userID : theNeighborhood) {
      possibleItemIDs.addAll(dataModel.getItemIDsFromUser(userID));
    }
    if (!includeKnownItems) {
      possibleItemIDs.removeAll(dataModel.getItemIDsFromUser(theUserID));
    }
    return possibleItemIDs;
View Full Code Here
38
39
40
41
42
43
44
45
46
47
48
  @Override
  protected FastIDSet doGetCandidateItems(long[] preferredItemIDs, DataModel dataModel, boolean includeKnownItems)
    throws TasteException {
    FastIDSet candidateItemIDs = new FastIDSet();
    for (long itemID : preferredItemIDs) {
      candidateItemIDs.addAll(similarity.allSimilarItemIDs(itemID));
    }
    if (!includeKnownItems) {
      candidateItemIDs.removeAll(preferredItemIDs);
    }
    return candidateItemIDs;
View Full Code Here
34
35
36
37
38
39
40
41
42
43
44
    FastIDSet possibleItemsIDs = new FastIDSet();
    for (long itemID : preferredItemIDs) {
      PreferenceArray itemPreferences = dataModel.getPreferencesForItem(itemID);
      int numUsersPreferringItem = itemPreferences.length();
      for (int index = 0; index < numUsersPreferringItem; index++) {
        possibleItemsIDs.addAll(dataModel.getItemIDsFromUser(itemPreferences.getUserID(index)));
      }
    }
    if (!includeKnownItems) {
      possibleItemsIDs.removeAll(preferredItemIDs);
    }
View Full Code Here
70
71
72
73
74
75
76
77
78
79
80
  }

  protected FastIDSet doGetCandidateItemsInternal(long[] validItemIDs,
    long[] seenItemIDs) throws TasteException {
      FastIDSet possibleItemsIDs = new FastIDSet();
      possibleItemsIDs.addAll(this.validItemIDs);

      if (seenItemIDs != null)
        possibleItemsIDs.removeAll(seenItemIDs);

      return possibleItemsIDs;
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.