GitPlanet

Cheap and reliable Node.js hosting starts at $3/month, and $1/month static HTML hosting

Created with love in Canada, visit hostnodejs.com today

Feel like to post an Ad? Learn Details
All Projects → gasevi → Pyreclab

gasevi / Pyreclab

Licence: gpl-3.0
pyRecLab is a library for quickly testing and prototyping of traditional recommender system methods, such as User KNN, Item KNN and FunkSVD Collaborative Filtering. It is developed and maintained by Gabriel Sepúlveda and Vicente Domínguez, advised by Prof. Denis Parra, all of them in Computer Science Department at PUC Chile, IA Lab and SocVis Lab.

Labels

ratingsvd

Projects that are alternatives of or similar to Pyreclab

Smileyrating
SmileyRating is a simple rating bar for android. It displays animated smileys as rating icon.
Stars: ✭ 1,038 (+861.11%)
Mutual labels:  rating
Laravel Review Rateable
Review & Rating System For Lavavel 5, 6 & 7
Stars: ✭ 72 (-33.33%)
Mutual labels:  rating
Ristretto
Randomized Dimension Reduction Library
Stars: ✭ 92 (-14.81%)
Mutual labels:  svd
Deeplearning Mxnet
MXNet for CTR
Stars: ✭ 51 (-52.78%)
Mutual labels:  svd
Recommender
A recommendation system using tensorflow
Stars: ✭ 69 (-36.11%)
Mutual labels:  svd
Trustgraph
Decentralized trust ratings using signed claims
Stars: ✭ 75 (-30.56%)
Mutual labels:  rating
Genericsvd.jl
Singular Value Decomposition for generic number types
Stars: ✭ 40 (-62.96%)
Mutual labels:  svd
Recommenders
Best Practices on Recommendation Systems
Stars: ✭ 11,818 (+10842.59%)
Mutual labels:  rating
Mgstarratingview
MGStarRatingView is a view for rating.
Stars: ✭ 70 (-35.19%)
Mutual labels:  rating
Rating
rating system for laravel 5
Stars: ✭ 85 (-21.3%)
Mutual labels:  rating
Laravel Reviewable
Adds a reviewable feature to your laravel app.
Stars: ✭ 57 (-47.22%)
Mutual labels:  rating
Simpleratingbar
A simple RatingBar that you can easier to customize image and animations
Stars: ✭ 1,157 (+971.3%)
Mutual labels:  rating
Ml code
A repository for recording the machine learning code
Stars: ✭ 75 (-30.56%)
Mutual labels:  svd
Yii2 Widget Rating
A Yii2 widget for the simple yet powerful bootstrap-star-rating plugin with fractional rating support (sub repo split from yii2-widgets)
Stars: ✭ 47 (-56.48%)
Mutual labels:  rating
Laravel Likeable
Rate Eloquent models with Likes and Dislikes in Laravel. Development moved to Laravel Love package!
Stars: ✭ 95 (-12.04%)
Mutual labels:  rating
Ng Bootstrap
Angular powered Bootstrap
Stars: ✭ 7,872 (+7188.89%)
Mutual labels:  rating
Recommend
Python 3.6 下的推荐算法解析，尽量使用简单的语言剖析原理，相似度度量、协同过滤、矩阵分解等
Stars: ✭ 72 (-33.33%)
Mutual labels:  svd
Torch Batch Svd
A 100x faster SVD for PyTorch⚡️
Stars: ✭ 107 (-0.93%)
Mutual labels:  svd
Arcchartview
Arc Chart View (Draw Creative Statistic Arc Charts)
Stars: ✭ 96 (-11.11%)
Mutual labels:  rating
Starrate
swift电商五星评价，星星评分控件，支持自定义数量、拖拽、间隔、设置最小星星数等操作
Stars: ✭ 85 (-21.3%)
Mutual labels:  rating
View All Similar Projects ➔

pyRecLab: Recommendation lab for Python

Overview

pyRecLab is a recommendation library designed for training recommendation models with a friendly and easy-to-use interface, keeping a good performance in memory and CPU usage.

In order to achieve this, pyRecLab is built as a Python module to give a friendly access to its algorithms and it is completely developed in C++ to avoid the lack of performace of the interpreted languages.

At this moment, the following recommendation algorithms are supported:

RecSys Algorithm Rating Prediction Item Recommendation Implicit Feedback
User Average x x
Item Average x x
Slope One x x
User Based KNN x x
Item Based KNN x x
Funk's SVD x x
Most Popular x
ALS x x
ALS with Conjugate Gradient x x
BPR for Matrix Factorization x x

Although pyRecLab can be compiled on most popular operating system, it has been tested on the following distributions.

Operating System Version
Ubuntu 16.04
CentOS 6.4
Mac OS X 10.11 ( El Capitan )
Mac OS X 10.12 ( Sierra )

Citations

If you use this library, please cite:

@inproceedings{1706.06291v2, author = {Gabriel Sepulveda and Vicente Dominguez and Denis Parra}, title = {pyRecLab: A Software Library for Quick Prototyping of Recommender Systems}, year = {2017}, month = {August}, eprint = {arXiv:1706.06291v2}, keywords = {Recommender Systems, Software Development, Recommender Library, Python Library} }

Check out our paper

Install

pyRecLab can be installed directly using pip as follow:

Python 2

pip install pyreclab

Python 3

pip3 install pyreclab

Build and Install

In case there is not a package available for your favorite operating system, you can build pyRecLab through the following steps:

1.- Before starting, verify you have libpython-dev, boost and cmake installed on your system. If not, install it through your distribution's package manager, as shown next.

Debian based OS's ( Ubuntu )

$ sudo apt-get install libpython-dev
$ sudo apt-get install cmake
$ sudo apt-get install libboost-dev

Note: for Python 3.x, install libpython3-dev instead of libpython-dev.

RedHat based OS's ( CentOS )

$ yum install python-devel
$ yum install cmake
$ yum install boost-devel

MAC OS X

$ brew install cmake
$ brew install boost

2.- Clone the source code of pyRecLab in a local directory.

$ git clone https://github.com/gasevi/pyreclab.git

3.- Build the Python module ( default: Python 2.7 ).

$ cd pyreclab
$ cmake .
$ make

By default, pyRecLab will be compiled for Python 2.7. If you want to build it for Python 3.x, you can execute the following steps:

$ cd pyreclab
$ cmake -DCMAKE_PYTHON_VERSION=3.x .
$ make

4.- Install pyRecLab.

$ sudo make install

API Documentation

pyRecLab provides the following classes for representing each of the recommendation algorithm currenly supported:

>>> from pyreclab import <RecAlg>

or import the entire module as you prefer

>>> import pyreclab

Due to the different nature of each algorithm, their implementations can have some variations on its parameters. For this reason, each class is described in detail in the following sections.

pyreclab.UserAvg

  • Instance creation
>>> obj = pyreclab.UserAvg( dataset = filename,
                            dlmchar = b'\t',
                            header = False,
                            usercol = 0,
                            itemcol = 1,
                            ratingcol = 2 )
Parameter Type Default value Description
dataset mandatory N.A. Dataset filename with fields: userid, itemid and rating
dlmchar optional tab Delimiter character between fields (userid, itemid, rating)
header optional False Whether dataset filename contains a header line to skip
usercol optional 0 User column position in dataset file
itemcol optional 1 Item column position in dataset file
ratingcol optional 2 Rating column position in dataset file
  • Training
>>> obj.train( progress = False )
Parameter Type Default value Description
progress optional False Show progress bar
  • Rating prediction
>>> prediction = obj.predict( userId, itemId )
Parameter Type Default value Description
userId mandatory N.A. User identifier
itemId mandatory N.A. Item identifier
  • Top-N item recommendation
>>> ranking = obj.recommend( userId, topN, includeRated )
Parameter Type Default value Description
userId mandatory N.A. User identifier
topN optional 10 Top N items to recommend
includeRated optional False Include rated items in ranking generation
  • Testing and evaluation for prediction
>>> predictionList, mae, rmse = obj.test( input_file = testset,
                                          dlmchar = b'\t',
                                          header = False,
                                          usercol = 0,
                                          itemcol = 1,
                                          ratingcol = 2,
                                          output_file = 'predictions.csv' )
Parameter Type Default value Description
input_file mandatory N.A. Testset filename
dlmchar optional tab Delimiter character between fields (userid, itemid, rating)
header optional False Dataset filename contains first line header to skip
usercol optional 0 User column position in dataset file
itemcol optional 1 Item column position in dataset file
ratingcol optional 2 Rating column position in dataset file
output_file optional N.A. Output file to write predictions
  • Testing for recommendation
>>> recommendationList, map, ndcg = obj.testrec( input_file = testset,
                                                 dlmchar = b'\t',
                                                 header = False,
                                                 usercol = 0,
                                                 itemcol = 1,
                                                 ratingcol = 2,
                                                 topn = 10,
                                                 output_file = 'ranking.json',
                                                 relevance_threshold = 0,
                                                 includeRated = False )
Parameter Type Default value Description
input_file mandatory N.A. Testset filename
dlmchar optional tab Delimiter character between fields (userid, itemid, rating)
header optional False Dataset filename contains first line header to skip
usercol optional 0 User column position in dataset file
itemcol optional 1 Item column position in dataset file
ratingcol optional 2 Rating column position in dataset file
topn optional 10 Top N items to recommend
output_file optional N.A. Output file to write predictions
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
includeRated optional False Include rated items in ranking generation
  • Precision
>>> precision = obj.precision( user_id,
                               retrieved,
                               topn = 10,
                               relevance_threshold = 0,
                               include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Recall
>>> recall = obj.recall( user_id,
                         retrieved,
                         topn = 10,
                         relevance_threshold = 0,
                         include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Area Under the ROC Curve (AUC)
>>> auc = obj.AUC( user_id,
                   retrieved,
                   topn = 10,
                   relevance_threshold = 0,
                   include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Mean Average precision
>>> map = obj.MAP( user_id,
                   retrieved,
                   topn = 10,
                   relevance_threshold = 0,
                   include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Normalized Discounted Cumulative Gain
>>> map = obj.nDCG( user_id,
                    retrieved,
                    topn = 10,
                    relevance_threshold = 0,
                    include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation

pyreclab.ItemAvg

  • Instance creation
>>> obj = pyreclab.ItemAvg( dataset = filename,
                            dlmchar = b'\t',
                            header = False,
                            usercol = 0,
                            itemcol = 1,
                            ratingcol = 2 )
Parameter Type Default value Description
dataset mandatory N.A. Dataset filename with fields: userid, itemid and rating
dlmchar optional tab Delimiter character between fields (userid, itemid, rating)
header optional False Whether dataset filename contains a header line to skip
usercol optional 0 User column position in dataset file
itemcol optional 1 Item column position in dataset file
ratingcol optional 2 Rating column position in dataset file
  • Training
>>> obj.train( progress = False )
Parameter Type Default value Description
progress optional False Show progress bar
  • Rating prediction
>>> prediction = obj.predict( userId, itemId )
Parameter Type Default value Description
userId mandatory N.A. User identifier
itemId mandatory N.A. Item identifier
  • Top-N item recommendation
>>> ranking = obj.recommend( userId, topN, includeRated )
Parameter Type Default value Description
userId mandatory N.A. User identifier
topN optional 10 Top N items to recommend
includeRated optional False Include rated items in ranking generation
  • Testing and evaluation for prediction
>>> predictionList, mae, rmse = obj.test( input_file = testset,
                                          dlmchar = b'\t',
                                          header = False,
                                          usercol = 0,
                                          itemcol = 1,
                                          ratingcol = 2,
                                          output_file = 'predictions.csv' )
Parameter Type Default value Description
input_file mandatory N.A. Testset filename
dlmchar optional tab Delimiter character between fields (userid, itemid, rating)
header optional False Dataset filename contains first line header to skip
usercol optional 0 User column position in dataset file
itemcol optional 1 Item column position in dataset file
ratingcol optional 2 Rating column position in dataset file
output_file optional N.A. Output file to write predictions
  • Testing for recommendation
>>> recommendationList, map, ndcg = obj.testrec( input_file = testset,
                                                 dlmchar = b'\t',
                                                 header = False,
                                                 usercol = 0,
                                                 itemcol = 1,
                                                 ratingcol = 2,
                                                 topn = 10,
                                                 output_file = 'ranking.json',
                                                 relevance_threshold = 2,
                                                 includeRated = False )
Parameter Type Default value Description
input_file mandatory N.A. Testset filename
dlmchar optional tab Delimiter character between fields (userid, itemid, rating)
header optional False Dataset filename contains first line header to skip
usercol optional 0 User column position in dataset file
itemcol optional 1 Item column position in dataset file
ratingcol optional 2 Rating column position in dataset file
topn optional 10 Top N items to recommend
output_file optional N.A. Output file to write predictions
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
includeRated optional False Include rated items in ranking generation
  • Precision
>>> precision = obj.precision( user_id,
                               retrieved,
                               topn = 10,
                               relevance_threshold = 0,
                               include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Recall
>>> recall = obj.recall( user_id,
                         retrieved,
                         topn = 10,
                         relevance_threshold = 0,
                         include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Area Under the ROC Curve (AUC)
>>> auc = obj.AUC( user_id,
                   retrieved,
                   topn = 10,
                   relevance_threshold = 0,
                   include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Mean Average precision
>>> map = obj.MAP( user_id,
                   retrieved,
                   topn = 10,
                   relevance_threshold = 0,
                   include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Normalized Discounted Cumulative Gain
>>> map = obj.nDCG( user_id,
                    retrieved,
                    topn = 10,
                    relevance_threshold = 0,
                    include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation

pyreclab.SlopeOne

  • Instance creation
>>> obj = pyreclab.SlopeOne( dataset = filename,
                             dlmchar = b'\t',
                             header = False,
                             usercol = 0,
                             itemcol = 1,
                             ratingcol = 2 )
Parameter Type Default value Description
dataset mandatory N.A. Dataset filename with fields: userid, itemid and rating
dlmchar optional tab Delimiter character between fields (userid, itemid, rating)
header optional False Whether dataset filename contains a header line to skip
usercol optional 0 User column position in dataset file
itemcol optional 1 Item column position in dataset file
ratingcol optional 2 Rating column position in dataset file
  • Training
>>> obj.train( progress = False )
Parameter Type Default value Description
progress optional False Show progress bar
  • Rating prediction
prediction = obj.predict( userId, itemId )
Parameter Type Default value Description
userId mandatory N.A. User identifier
itemId mandatory N.A. Item identifier
  • Top-N item recommendation
>>> ranking = obj.recommend( userId, topN, includeRated )
Parameter Type Default value Description
userId mandatory N.A. User identifier
topN optional 10 Top N items to recommend
includeRated optional False Include rated items in ranking generation
  • Testing and evaluation for prediction
>>> predictionList, mae, rmse = obj.test( input_file = testset,
                                          dlmchar = b'\t',
                                          header = False,
                                          usercol = 0,
                                          itemcol = 1,
                                          ratingcol = 2,
                                          output_file = 'predictions.csv' )
Parameter Type Default value Description
input_file mandatory N.A. Testset filename
dlmchar optional tab Delimiter character between fields (userid, itemid, rating)
header optional False Dataset filename contains first line header to skip
usercol optional 0 User column position in dataset file
itemcol optional 1 Item column position in dataset file
ratingcol optional 2 Rating column position in dataset file
output_file optional N.A. Output file to write predictions
  • Testing for recommendation
>>> recommendationList, map, ndcg = obj.testrec( input_file = testset,
                                                 dlmchar = b'\t',
                                                 header = False,
                                                 usercol = 0,
                                                 itemcol = 1,
                                                 ratingcol = 2,
                                                 topn = 10,
                                                 output_file = 'ranking.json',
                                                 relevance_threshold = 2,
                                                 includeRated = False )
Parameter Type Default value Description
input_file mandatory N.A. Testset filename
dlmchar optional tab Delimiter character between fields (userid, itemid, rating)
header optional False Dataset filename contains first line header to skip
usercol optional 0 User column position in dataset file
itemcol optional 1 Item column position in dataset file
ratingcol optional 2 Rating column position in dataset file
topn optional 10 Top N items to recommend
output_file optional N.A. Output file to write predictions
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
includeRated optional False Include rated items in ranking generation
  • Precision
>>> precision = obj.precision( user_id,
                               retrieved,
                               topn = 10,
                               relevance_threshold = 0,
                               include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Recall
>>> recall = obj.recall( user_id,
                         retrieved,
                         topn = 10,
                         relevance_threshold = 0,
                         include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Area Under the ROC Curve (AUC)
>>> auc = obj.AUC( user_id,
                   retrieved,
                   topn = 10,
                   relevance_threshold = 0,
                   include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Mean Average precision
>>> map = obj.MAP( user_id,
                   retrieved,
                   topn = 10,
                   relevance_threshold = 0,
                   include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Normalized Discounted Cumulative Gain
>>> map = obj.nDCG( user_id,
                    retrieved,
                    topn = 10,
                    relevance_threshold = 0,
                    include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation

pyreclab.UserKnn

  • Instance creation
>>> obj = pyreclab.UserKnn( dataset = filename,
                            dlmchar = b'\t',
                            header = False,
                            usercol = 0,
                            itemcol = 1,
                            ratingcol = 2 )
Parameter Type Default value Description
dataset mandatory N.A. Dataset filename with fields: userid, itemid and rating
dlmchar optional tab Delimiter character between fields (userid, itemid, rating)
header optional False Whether dataset filename contains a header line to skip
usercol optional 0 User column position in dataset file
itemcol optional 1 Item column position in dataset file
ratingcol optional 2 Rating column position in dataset file
  • Training
>>> obj.train( knn, similarity, progress = False )
Parameter Type Default value Valid values Description
knn optional 10 positive integer K nearest neighbors
similarity optional 'pearson' 'pearson', 'cosine' Similarity metric
progress optional False Show progress bar
  • Rating prediction
>>> prediction = obj.predict( userId, itemId )
Parameter Type Default value Description
userId mandatory N.A. User identifier
itemId mandatory N.A. Item identifier
  • Top-N item recommendation
>>> ranking = obj.recommend( userId, topN, includeRated )
Parameter Type Default value Description
userId mandatory N.A. User identifier
topN optional 10 Top N items to recommend
includeRated optional False Include rated items in ranking generation
  • Testing and evaluation for prediction
>>> predictionList, mae, rmse = obj.test( input_file = testset,
                                          dlmchar = b'\t',
                                          header = False,
                                          usercol = 0,
                                          itemcol = 1,
                                          ratingcol = 2,
                                          output_file = 'predictions.csv' )
Parameter Type Default value Description
input_file mandatory N.A. Testset filename
dlmchar optional tab Delimiter character between fields (userid, itemid, rating)
header optional False Dataset filename contains first line header to skip
usercol optional 0 User column position in dataset file
itemcol optional 1 Item column position in dataset file
ratingcol optional 2 Rating column position in dataset file
output_file optional N.A. Output file to write predictions
  • Testing for recommendation
>>> recommendationList, map, ndcg = obj.testrec( input_file = testset,
                                                 dlmchar = b'\t',
                                                 header = False,
                                                 usercol = 0,
                                                 itemcol = 1,
                                                 ratingcol = 2,
                                                 topn = 10,
                                                 output_file = 'ranking.json',
                                                 relevance_threshold = 2,
                                                 includeRated = False )
Parameter Type Default value Description
input_file mandatory N.A. Testset filename
dlmchar optional tab Delimiter character between fields (userid, itemid, rating)
header optional False Dataset filename contains first line header to skip
usercol optional 0 User column position in dataset file
itemcol optional 1 Item column position in dataset file
ratingcol optional 2 Rating column position in dataset file
topn optional 10 Top N items to recommend
output_file optional N.A. Output file to write predictions
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
includeRated optional False Include rated items in ranking generation
  • Precision
>>> precision = obj.precision( user_id,
                               retrieved,
                               topn = 10,
                               relevance_threshold = 0,
                               include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Recall
>>> recall = obj.recall( user_id,
                         retrieved,
                         topn = 10,
                         relevance_threshold = 0,
                         include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Area Under the ROC Curve (AUC)
>>> auc = obj.AUC( user_id,
                   retrieved,
                   topn = 10,
                   relevance_threshold = 0,
                   include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Mean Average precision
>>> map = obj.MAP( user_id,
                   retrieved,
                   topn = 10,
                   relevance_threshold = 0,
                   include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Normalized Discounted Cumulative Gain
>>> map = obj.nDCG( user_id,
                    retrieved,
                    topn = 10,
                    relevance_threshold = 0,
                    include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation

pyreclab.ItemKnn

  • Instance creation
>>> obj = pyreclab.ItemKnn( dataset = filename,
                            dlmchar = b'\t',
                            header = False,
                            usercol = 0,
                            itemcol = 1,
                            ratingcol = 2 )
Parameter Type Default value Description
dataset mandatory N.A. Dataset filename with fields: userid, itemid and rating
dlmchar optional tab Delimiter character between fields (userid, itemid, rating)
header optional False Whether dataset filename contains a header line to skip
usercol optional 0 User column position in dataset file
itemcol optional 1 Item column position in dataset file
ratingcol optional 2 Rating column position in dataset file
  • Training
>>> obj.train( knn, similarity, progress = False )
Parameter Type Default value Valid values Description
knn optional 10 positive integer K nearest neighbors
similarity optional 'pearson' 'pearson', 'cosine' Similarity metric
progress optional False Show progress bar
  • Rating prediction
>>> prediction = obj.predict( userId, itemId )
Parameter Type Default value Description
userId mandatory N.A. User identifier
itemId mandatory N.A. Item identifier
  • Top-N item recommendation
>>> ranking = obj.recommend( userId, topN, includeRated )
Parameter Type Default value Description
userId mandatory N.A. User identifier
topN optional 10 Top N items to recommend
includeRated optional False Include rated items in ranking generation
  • Testing and evaluation for prediction
>>> predictionList, mae, rmse = obj.test( input_file = testset,
                                          dlmchar = b'\t',
                                          header = False,
                                          usercol = 0,
                                          itemcol = 1,
                                          ratingcol = 2,
                                          output_file = 'predictions.csv' )
Parameter Type Default value Description
input_file mandatory N.A. Testset filename
dlmchar optional tab Delimiter character between fields (userid, itemid, rating)
header optional False Dataset filename contains first line header to skip
usercol optional 0 User column position in dataset file
itemcol optional 1 Item column position in dataset file
ratingcol optional 2 Rating column position in dataset file
output_file optional N.A. Output file to write predictions
  • Testing for recommendation
>>> recommendationList, map, ndcg = obj.testrec( input_file = testset,
                                                 dlmchar = b'\t',
                                                 header = False,
                                                 usercol = 0,
                                                 itemcol = 1,
                                                 ratingcol = 2,
                                                 topn = 10,
                                                 output_file = 'ranking.json',
                                                 relevance_threshold = 2,
                                                 includeRated = False )
Parameter Type Default value Description
input_file mandatory N.A. Testset filename
dlmchar optional tab Delimiter character between fields (userid, itemid, rating)
header optional False Dataset filename contains first line header to skip
usercol optional 0 User column position in dataset file
itemcol optional 1 Item column position in dataset file
ratingcol optional 2 Rating column position in dataset file
topn optional 10 Top N items to recommend
output_file optional N.A. Output file to write predictions
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
includeRated optional False Include rated items in ranking generation
  • Precision
>>> precision = obj.precision( user_id,
                               retrieved,
                               topn = 10,
                               relevance_threshold = 0,
                               include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Recall
>>> recall = obj.recall( user_id,
                         retrieved,
                         topn = 10,
                         relevance_threshold = 0,
                         include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Area Under the ROC Curve (AUC)
>>> auc = obj.AUC( user_id,
                   retrieved,
                   topn = 10,
                   relevance_threshold = 0,
                   include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Mean Average precision
>>> map = obj.MAP( user_id,
                   retrieved,
                   topn = 10,
                   relevance_threshold = 0,
                   include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Normalized Discounted Cumulative Gain
>>> map = obj.nDCG( user_id,
                    retrieved,
                    topn = 10,
                    relevance_threshold = 0,
                    include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation

pyreclab.SVD

  • Instance creation
>>> obj = pyreclab.SVD( factors = 1000,
                        dataset = filename,
                        dlmchar = b'\t',
                        header = False,
                        usercol = 0,
                        itemcol = 1,
                        ratingcol = 2 )
Parameter Type Default value Description
factors optional 1000 Number of latent factors in matrix factorization
dataset mandatory N.A. Dataset filename with fields: userid, itemid and rating
dlmchar optional tab Delimiter character between fields (userid, itemid, rating)
header optional False Whether dataset filename contains a header line to skip
usercol optional 0 User column position in dataset file
itemcol optional 1 Item column position in dataset file
ratingcol optional 2 Rating column position in dataset file
  • Training
>>> obj.train( maxiter = 100, lr = 0.01, lamb = 0.1, progress = False )
Parameter Type Default value Description
maxiter optional 100 Maximum number of iterations reached without convergence
lr optional 0.01 Learning rate
lamb optional 0.1 Regularization parameter
progress optional False Show progress bar
  • Rating prediction
>>> prediction = obj.predict( userId, itemId )
Parameter Type Default value Description
userId mandatory N.A. User identifier
itemId mandatory N.A. Item identifier
  • Top-N item recommendation
>>> ranking = obj.recommend( userId, topN, includeRated )
Parameter Type Default value Description
userId mandatory N.A. User identifier
topN optional 10 Top N items to recommend
includeRated optional False Include rated items in ranking generation
  • Testing and evaluation for prediction
>>> predictionList, mae, rmse = obj.test( input_file = testset,
                                          dlmchar = b'\t',
                                          header = False,
                                          usercol = 0,
                                          itemcol = 1,
                                          ratingcol = 2,
                                          output_file = 'predictions.csv' )
Parameter Type Default value Description
input_file mandatory N.A. Testset filename
dlmchar optional tab Delimiter character between fields (userid, itemid, rating)
header optional False Dataset filename contains first line header to skip
usercol optional 0 User column position in dataset file
itemcol optional 1 Item column position in dataset file
ratingcol optional 2 Rating column position in dataset file
output_file optional N.A. Output file to write predictions
  • Testing for recommendation
>>> recommendationList, map, ndcg = obj.testrec( input_file = testset,
                                                 dlmchar = b'\t',
                                                 header = False,
                                                 usercol = 0,
                                                 itemcol = 1,
                                                 ratingcol = 2,
                                                 topn = 10,
                                                 output_file = 'ranking.json',
                                                 relevance_threshold = 2,
                                                 includeRated = False )
Parameter Type Default value Description
input_file mandatory N.A. Testset filename
dlmchar optional tab Delimiter character between fields (userid, itemid, rating)
header optional False Dataset filename contains first line header to skip
usercol optional 0 User column position in dataset file
itemcol optional 1 Item column position in dataset file
ratingcol optional 2 Rating column position in dataset file
topn optional 10 Top N items to recommend
output_file optional N.A. Output file to write predictions
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
includeRated optional False Include rated items in ranking generation
  • Precision
>>> precision = obj.precision( user_id,
                               retrieved,
                               topn = 10,
                               relevance_threshold = 0,
                               include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Recall
>>> recall = obj.recall( user_id,
                         retrieved,
                         topn = 10,
                         relevance_threshold = 0,
                         include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Area Under the ROC Curve (AUC)
>>> auc = obj.AUC( user_id,
                   retrieved,
                   topn = 10,
                   relevance_threshold = 0,
                   include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Mean Average precision
>>> map = obj.MAP( user_id,
                   retrieved,
                   topn = 10,
                   relevance_threshold = 0,
                   include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Normalized Discounted Cumulative Gain
>>> map = obj.nDCG( user_id,
                    retrieved,
                    topn = 10,
                    relevance_threshold = 0,
                    include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Loss
>>> current_loss = obj.loss()
  • Reset factors
>>> obj.reset()

pyreclab.MostPopular

  • Instance creation
>>> obj = pyreclab.MostPopular( dataset = filename,
                                dlmchar = b'\t',
                                header = False,
                                usercol = 0,
                                itemcol = 1,
                                ratingcol = 2 )
Parameter Type Default value Description
dataset mandatory N.A. Dataset filename with fields: userid, itemid and rating
dlmchar optional tab Delimiter character between fields (userid, itemid, rating)
header optional False Whether dataset filename contains a header line to skip
usercol optional 0 User column position in dataset file
itemcol optional 1 Item column position in dataset file
ratingcol optional 2 Rating column position in dataset file
  • Training
>>> obj.train( progress = False )
Parameter Type Default value Description
progress optional False Show progress bar
  • Top-N item recommendation
>>> ranking = obj.recommend( userId, topN, includeRated )
Parameter Type Default value Description
userId mandatory N.A. User identifier
topN optional 10 Top N items to recommend
includeRated optional False Include rated items in ranking generation
  • Testing for recommendation
>>> recommendationList, map, ndcg = obj.testrec( input_file = testset,
                                                 dlmchar = b'\t',
                                                 header = False,
                                                 usercol = 0,
                                                 itemcol = 1,
                                                 ratingcol = 2,
                                                 topn = 10,
                                                 output_file = 'ranking.json',
                                                 relevance_threshold = 2,
                                                 includeRated = False )
Parameter Type Default value Description
input_file mandatory N.A. Testset filename
dlmchar optional tab Delimiter character between fields (userid, itemid, rating)
header optional False Dataset filename contains first line header to skip
usercol optional 0 User column position in dataset file
itemcol optional 1 Item column position in dataset file
ratingcol optional 2 Rating column position in dataset file
output_file optional N.A. Output file to write rankings
topN optional 10 Top N items to recommend
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
includeRated optional False Include rated items in ranking generation
  • Precision
>>> precision = obj.precision( user_id,
                               retrieved,
                               topn = 10,
                               relevance_threshold = 0,
                               include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Recall
>>> recall = obj.recall( user_id,
                         retrieved,
                         topn = 10,
                         relevance_threshold = 0,
                         include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Area Under the ROC Curve (AUC)
>>> auc = obj.AUC( user_id,
                   retrieved,
                   topn = 10,
                   relevance_threshold = 0,
                   include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Mean Average precision
>>> map = obj.MAP( user_id,
                   retrieved,
                   topn = 10,
                   relevance_threshold = 0,
                   include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Normalized Discounted Cumulative Gain
>>> map = obj.nDCG( user_id,
                    retrieved,
                    topn = 10,
                    relevance_threshold = 0,
                    include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation

pyreclab.IFAls

  • Instance creation
>>> obj = pyreclab.IFAls( factors = 50,
                          dataset = filename,
                          dlmchar = b'\t',
                          header = False,
                          usercol = 0,
                          itemcol = 1,
                          observationcol = 2 )
Parameter Type Default value Description
factors optional 50 Number of latent factors in matrix factorization
dataset mandatory N.A. Dataset filename with fields: userid, itemid and rating
dlmchar optional tab Delimiter character between fields (userid, itemid, rating)
header optional False Whether dataset filename contains a header line to skip
usercol optional 0 User column position in dataset file
itemcol optional 1 Item column position in dataset file
observationcol optional 2 Observation column position in dataset file
  • Training
>>> obj.train( maxiter, lambd, progress = False )
Parameter Type Default value Description
alsNumIter optional 5 Number of iterations in ALS algorithm
lambd optional 10 Regularization parameter
progress optional False Show progress bar
  • Top-N item recommendation
>>> ranking = obj.recommend( userId, topN, includeRated )
Parameter Type Default value Description
userId mandatory N.A. User identifier
topN optional 10 Top N items to recommend
includeRated optional False Include rated items in ranking generation
  • Testing for recommendation
>>> recommendationList, map, ndcg = obj.testrec( input_file = testset,
                                                 dlmchar = b'\t',
                                                 header = False,
                                                 usercol = 0,
                                                 itemcol = 1,
                                                 ratingcol = 2,
                                                 topn = 10,
                                                 output_file = 'ranking.json',
                                                 relevance_threshold = 2,
                                                 includeRated = False )
Parameter Type Default value Description
input_file mandatory N.A. Testset filename
dlmchar optional tab Delimiter character between fields (userid, itemid, rating)
header optional False Dataset filename contains first line header to skip
usercol optional 0 User column position in dataset file
itemcol optional 1 Item column position in dataset file
ratingcol optional 2 Rating column position in dataset file
output_file optional N.A. Output file to write rankings
topN optional 10 Top N items to recommend
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
includeRated optional False Include rated items in ranking generation
  • Precision
>>> precision = obj.precision( user_id,
                               retrieved,
                               topn = 10,
                               relevance_threshold = 0,
                               include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Recall
>>> recall = obj.recall( user_id,
                         retrieved,
                         topn = 10,
                         relevance_threshold = 0,
                         include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Area Under the ROC Curve (AUC)
>>> auc = obj.AUC( user_id,
                   retrieved,
                   topn = 10,
                   relevance_threshold = 0,
                   include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Mean Average precision
>>> map = obj.MAP( user_id,
                   retrieved,
                   topn = 10,
                   relevance_threshold = 0,
                   include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Normalized Discounted Cumulative Gain
>>> map = obj.nDCG( user_id,
                    retrieved,
                    topn = 10,
                    relevance_threshold = 0,
                    include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Reset factors
>>> obj.reset()

pyreclab.IFAlsConjugateGradient

  • Instance creation
>>> obj = pyreclab.IFAlsConjugateGradient( factors = 50,
                                           dataset = filename,
                                           dlmchar = b'\t',
                                           header = False,
                                           usercol = 0,
                                           itemcol = 1,
                                           observationcol = 2 )
Parameter Type Default value Description
factors optional 50 Number of latent factors in matrix factorization
dataset mandatory N.A. Dataset filename with fields: userid, itemid and rating
dlmchar optional tab Delimiter character between fields (userid, itemid, rating)
header optional False Whether dataset filename contains a header line to skip
usercol optional 0 User column position in dataset file
itemcol optional 1 Item column position in dataset file
observationcol optional 2 Observation column position in dataset file
  • Training
>>> obj.train( alsNumIter, lambd, cgNumIter, progress = False )
Parameter Type Default value Description
alsNumIter optional 5 Number of iterations in ALS algorithm
lambd optional 10 Regularization parameter
cgNumIter optional 2 Number of iterations in Conjugate Gradient algorithm
progress optional False Show progress bar
  • Top-N item recommendation
>>> ranking = obj.recommend( userId, topN, includeRated )
Parameter Type Default value Description
userId mandatory N.A. User identifier
topN optional 10 Top N items to recommend
includeRated optional False Include rated items in ranking generation
  • Testing for recommendation
>>> recommendationList, map, ndcg = obj.testrec( input_file = testset,
                                                 dlmchar = b'\t',
                                                 header = False,
                                                 usercol = 0,
                                                 itemcol = 1,
                                                 ratingcol = 2,
                                                 topn = 10,
                                                 output_file = 'ranking.json',
                                                 relevance_threshold = 2,
                                                 includeRated = False )
Parameter Type Default value Description
input_file mandatory N.A. Testset filename
dlmchar optional tab Delimiter character between fields (userid, itemid, rating)
header optional False Dataset filename contains first line header to skip
usercol optional 0 User column position in dataset file
itemcol optional 1 Item column position in dataset file
ratingcol optional 2 Rating column position in dataset file
output_file optional N.A. Output file to write rankings
topN optional 10 Top N items to recommend
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
includeRated optional False Include rated items in ranking generation
  • Precision
>>> precision = obj.precision( user_id,
                               retrieved,
                               topn = 10,
                               relevance_threshold = 0,
                               include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Recall
>>> recall = obj.recall( user_id,
                         retrieved,
                         topn = 10,
                         relevance_threshold = 0,
                         include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Area Under the ROC Curve (AUC)
>>> auc = obj.AUC( user_id,
                   retrieved,
                   topn = 10,
                   relevance_threshold = 0,
                   include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Mean Average precision
>>> map = obj.MAP( user_id,
                   retrieved,
                   topn = 10,
                   relevance_threshold = 0,
                   include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Normalized Discounted Cumulative Gain
>>> map = obj.nDCG( user_id,
                    retrieved,
                    topn = 10,
                    relevance_threshold = 0,
                    include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Reset factors
>>> obj.reset()

pyreclab.BprMf

  • Instance creation
>>> obj = pyreclab.BprMf( factors = 20,
                          dataset = filename,
                          dlmchar = b'\t',
                          header = False,
                          usercol = 0,
                          itemcol = 1,
                          observationcol = 2 )
Parameter Type Default value Description
factors optional 20 Number of latent factors in matrix factorization
dataset mandatory N.A. Dataset filename with fields: userid, itemid and rating
dlmchar optional tab Delimiter character between fields (userid, itemid, rating)
header optional False Whether dataset filename contains a header line to skip
usercol optional 0 User column position in dataset file
itemcol optional 1 Item column position in dataset file
observationcol optional 2 Observation column position in dataset file
  • Training
>>> obj.train( maxiter = 100,
               lr = 0.1,
               lambda_w = 0.01,
               lambda_hp = 0.01,
               lambda_hm = 0.01,
               progress = True )
Parameter Type Default value Description
maxiter optional 100 Number of iterations
lr optional 0.1 Learning rate
lambda_w optional 0.01 Regularization parameter for the user features
lambda_hp optional 0.01 Regularization parameter for the item features and positive updates
lambda_hm optional 0.01 Regularization parameter for the item features and negative updates
progress optional False Show progress bar
  • Top-N item recommendation
>>> ranking = obj.recommend( userId, topN, includeRated )
Parameter Type Default value Description
userId mandatory N.A. User identifier
topN optional 10 Top N items to recommend
includeRated optional False Include rated items in ranking generation
  • Testing for recommendation
>>> recommendationList, map, ndcg = obj.testrec( input_file = testset,
                                                 dlmchar = b'\t',
                                                 header = False,
                                                 usercol = 0,
                                                 itemcol = 1,
                                                 ratingcol = 2,
                                                 topn = 10,
                                                 output_file = 'ranking.json',
                                                 relevance_threshold = 2,
                                                 includeRated = False )
Parameter Type Default value Description
input_file mandatory N.A. Testset filename
dlmchar optional tab Delimiter character between fields (userid, itemid, rating)
header optional False Dataset filename contains first line header to skip
usercol optional 0 User column position in dataset file
itemcol optional 1 Item column position in dataset file
ratingcol optional 2 Rating column position in dataset file
output_file optional N.A. Output file to write rankings
topN optional 10 Top N items to recommend
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
includeRated optional False Include rated items in ranking generation
  • Precision
>>> precision = obj.precision( user_id,
                               retrieved,
                               topn = 10,
                               relevance_threshold = 0,
                               include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Recall
>>> recall = obj.recall( user_id,
                         retrieved,
                         topn = 10,
                         relevance_threshold = 0,
                         include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Area Under the ROC Curve (AUC)
>>> auc = obj.AUC( user_id,
                   retrieved,
                   topn = 10,
                   relevance_threshold = 0,
                   include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Mean Average precision
>>> map = obj.MAP( user_id,
                   retrieved,
                   topn = 10,
                   relevance_threshold = 0,
                   include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Normalized Discounted Cumulative Gain
>>> map = obj.nDCG( user_id,
                    retrieved,
                    topn = 10,
                    relevance_threshold = 0,
                    include_rated = False )
Parameter Type Default value Description
user_id mandatory N.A. User identifier
retrieved optional N.A. Recommendation list for user 'user_id'
topn optional 10 Top N items to recommend. If 'retrieved' is provided, this value will be set to 'retrieved' length
relevance_threshold optional 0 Lower threshold to consider an item as relevant ( threshold value included )
include_rated optional False Include rated items in ranking generation
  • Loss
>>> current_loss = obj.loss()
  • Reset factors
>>> obj.reset()

On roadmap

  • Add Windows support.
  • Multi-threading.
Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].