com.cloudera.oryx.app.serving.als.model

Class ALSServingModel

java.lang.Object

com.cloudera.oryx.app.serving.als.model.ALSServingModel

All Implemented Interfaces:

ServingModel

public final class ALSServingModel
extends Object
implements ServingModel

Contains all data structures needed to serve real-time requests for an ALS-based recommender.

Method Summary

Modifier and Type	Method and Description
Collection<String>	getAllItemIDs()
Collection<String>	getAllUserIDs()
int	getFeatures()
float	getFractionLoaded()
Map<String,Integer>	getItemCounts()
float[]	getItemVector(String item)
Set<String>	getKnownItems(String user)
List<Pair<String,float[]>>	getKnownItemVectorsForUser(String user)
int	getNumItems()
int	getNumUsers()
RescorerProvider	getRescorerProvider()
Map<String,Integer>	getUserCounts()
float[]	getUserVector(String user)
Solver	getYTYSolver()
boolean	isImplicit()
Stream<Pair<String,Double>>	topN(CosineDistanceSensitiveFunction scoreFn, ToDoubleObjDoubleBiFunction<String> rescoreFn, int howMany, Predicate<String> allowedPredicate)
String	toString()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Method Detail

getFeatures

public int getFeatures()

isImplicit

public boolean isImplicit()

getRescorerProvider

public RescorerProvider getRescorerProvider()

getUserVector

public float[] getUserVector(String user)

getItemVector

public float[] getItemVector(String item)

getKnownItems

public Set<String> getKnownItems(String user)

Parameters:

user - user to get known items for

Returns:

set of known items for the user (immutable, but thread-safe)

getUserCounts

public Map<String,Integer> getUserCounts()

Returns:

mapping of user IDs to count of items the user has interacted with

getItemCounts

public Map<String,Integer> getItemCounts()

Returns:

mapping of item IDs to count of users that have interacted with that item

getKnownItemVectorsForUser

public List<Pair<String,float[]>> getKnownItemVectorsForUser(String user)

Parameters:

user - user to get known item vectors for

Returns:

null if the user is not known to the model, or if there are no known items for the user

topN

public Stream<Pair<String,Double>> topN(CosineDistanceSensitiveFunction scoreFn,
                                        ToDoubleObjDoubleBiFunction<String> rescoreFn,
                                        int howMany,
                                        Predicate<String> allowedPredicate)

getAllUserIDs

public Collection<String> getAllUserIDs()

Returns:

all user IDs in the model

getAllItemIDs

public Collection<String> getAllItemIDs()

Returns:

all item IDs in the model

getYTYSolver

public Solver getYTYSolver()

Returns:

a Solver for use in solving systems involving YT*Y

getNumUsers

public int getNumUsers()

Returns:

number of users in the model

getNumItems

public int getNumItems()

Returns:

number of items in the model

getFractionLoaded

public float getFractionLoaded()

Specified by:

getFractionLoaded in interface ServingModel

Returns:

fraction of IDs that were expected to be in the model whose value has been loaded from an update

toString

public String toString()

Overrides:

toString in class Object