org.apache.mahout.cf.taste.example.netflix
Class NetflixDataModel
java.lang.Object
org.apache.mahout.cf.taste.example.netflix.NetflixDataModel
- All Implemented Interfaces:
- Serializable, Refreshable, DataModel
public final class NetflixDataModel
- extends Object
- implements DataModel
This is a DataModel
that reads the Netflix data set, as represented in its
unpacked file structure. Note that unpacking the distribution entails untarring the main
archive, then untarring the archive of training set data files.
- See Also:
- Serialized Form
NetflixDataModel
public NetflixDataModel(File dataDirectory,
boolean useSubset)
throws IOException
- Parameters:
dataDirectory
- root directory of Netflix data set, unpackeduseSubset
- if true, will use only a small fraction of the data -- may be useful
for quick experiments
- Throws:
IOException
getUserIDs
public LongPrimitiveIterator getUserIDs()
throws TasteException
- Specified by:
getUserIDs
in interface DataModel
- Throws:
TasteException
getPreferencesFromUser
public PreferenceArray getPreferencesFromUser(long id)
throws TasteException
- Specified by:
getPreferencesFromUser
in interface DataModel
- Throws:
TasteException
getItemIDsFromUser
public FastIDSet getItemIDsFromUser(long userID)
throws TasteException
- Specified by:
getItemIDsFromUser
in interface DataModel
- Throws:
TasteException
getItemIDs
public LongPrimitiveIterator getItemIDs()
throws TasteException
- Specified by:
getItemIDs
in interface DataModel
- Throws:
TasteException
getPreferencesForItem
public PreferenceArray getPreferencesForItem(long itemID)
throws TasteException
- Specified by:
getPreferencesForItem
in interface DataModel
- Throws:
TasteException
getPreferenceValue
public Float getPreferenceValue(long userID,
long itemID)
throws TasteException
- Specified by:
getPreferenceValue
in interface DataModel
- Throws:
TasteException
getPreferenceTime
public Long getPreferenceTime(long userID,
long itemID)
throws TasteException
- Specified by:
getPreferenceTime
in interface DataModel
- Throws:
TasteException
getNumItems
public int getNumItems()
throws TasteException
- Specified by:
getNumItems
in interface DataModel
- Throws:
TasteException
getNumUsers
public int getNumUsers()
throws TasteException
- Specified by:
getNumUsers
in interface DataModel
- Throws:
TasteException
getNumUsersWithPreferenceFor
public int getNumUsersWithPreferenceFor(long itemID)
throws TasteException
- Specified by:
getNumUsersWithPreferenceFor
in interface DataModel
- Throws:
TasteException
getNumUsersWithPreferenceFor
public int getNumUsersWithPreferenceFor(long itemID1,
long itemID2)
throws TasteException
- Specified by:
getNumUsersWithPreferenceFor
in interface DataModel
- Throws:
TasteException
setPreference
public void setPreference(long userID,
long itemID,
float value)
- Specified by:
setPreference
in interface DataModel
removePreference
public void removePreference(long userID,
long itemID)
- Specified by:
removePreference
in interface DataModel
refresh
public void refresh(Collection<Refreshable> alreadyRefreshed)
- Specified by:
refresh
in interface Refreshable
hasPreferenceValues
public boolean hasPreferenceValues()
- Specified by:
hasPreferenceValues
in interface DataModel
getMaxPreference
public float getMaxPreference()
- Specified by:
getMaxPreference
in interface DataModel
getMinPreference
public float getMinPreference()
- Specified by:
getMinPreference
in interface DataModel
toString
public String toString()
- Overrides:
toString
in class Object
Copyright © 2008-2012 The Apache Software Foundation. All Rights Reserved.