Recommender

base.py

#-*- coding:utf-8 -*-

"""
Base Recommender Models.
"""

# Authors: Marcel Caraciolo <[email protected]>
#          Bruno Melo <[email protected]>
# License: BSD Style.

from scikits.learn.base import BaseEstimator


class BaseRecommender(BaseEstimator):
    """
    Base Class for Recommenders that suggest items for users.

    Should not be used directly, use derived classes instead

    Attributes
    ----------
     model:  DataModel
          Defines the data model where data is fetched.

    """

    def __init__(self, model):
        self.model = model

    def recommend(self, user_ids, how_many, **params):
        '''
        Return a list of recommended items, ordered from most strongly
        recommend to least.

        Parameters
        ----------
        user_ids: array of shape [n_user_ids: int or string]
                 Users for which recommendations are to be computed.
        how_many: int
                 Desired number of recommendations
        rescorer:  function, optional
                 Rescoring function to apply before final list of
                 recommendations.

        '''
        raise NotImplementedError("BaseRecommender is an abstract class.")

    def estimate_preference(self, **params):
        '''
        Return an estimated preference if the user has not expressed a
        preference for the item, or else the user's actual preference for the
        item. If a preference cannot be estimated, returns None.
        '''
        raise NotImplementedError("BaseRecommender is an abstract class.")

    def all_other_items(self, user_ids, **params):
        '''
        Return all items in the `model` for which the user has not expressed
        the preference and could possibly be recommended to the user.

        Parameters
        ----------
        user_ids: array of shape [n_user_ids: int or string]
                 User for which recommendations are to be computed.
        '''
        raise NotImplementedError("BaseRecommender is an abstract class.")

    def set_preferences(self, user_ids, item_ids, values):
        '''
        Set a new preference of a user for a specific item with a certain
        magnitude.

        Parameters
        ----------
        user_ids: array of shape [n_user_ids: int or string]
                 User for which the preference will be updated.

        item_ids: array of shape [n_item_ids: int or string]
                 Item that will be updated.

        values: array of shape [n_item_ids: int or string]
              The new magnitude for the preference of a item_id from a
                user_id.

        '''
        self.model.set_preferences(user_ids, item_ids, values)

    def remove_preferences(self, user_ids, item_ids):
        '''
        Remove a preference of a user for a specific item

        Parameters
        ----------
        user_ids: array of shape [n_user_ids: int or string]
                 User for which recommendations are to be computed.
        item_ids:  array of shape [n_item_ids: int or string]
                 Item that will be removed the preference for the user_id.

        '''
        self.model.remove_preferences(user_ids, item_ids)