Cornac：推荐系统实验框架库

Cornac：推荐系统实验框架库

Cornac

Cornac is a comparative framework for multimodal recommender systems. It focuses on making it convenient to work with models leveraging auxiliary data (e.g., item descriptive text and image, social network, etc). Cornac enables fast experiments and straightforward implementations of new models. It is highly compatible with existing machine learning libraries (e.g., TensorFlow, PyTorch).

Quick Links

Website | Documentation | Tutorials | Examples | Models | Datasets | Paper | Preferred.AI

Installation

Currently, we are supporting python 3. There are several ways to install Cornac:

From PyPI (you may need a C++ compiler):

pip3 install cornac

From Anaconda:

conda install cornac -c conda-forge

From the GitHub source (for latest updates):

pip3 install Cythongit clone https://github.com/PreferredAI/cornac.gitcd cornacpython3 setup.py install

Note:

Additional dependencies required by models are listed here.

Some algorithm implementations use OpenMP to support multi-threading. For OSX users, in order to run those algorithms efficiently, you might need to install gcc from Homebrew to have an OpenMP compiler:

brew install gcc | brew link gcc

If you want to utilize your GPUs, you might consider:

TensorFlow installation instructions.PyTorch installation instructions.cuDNN (for Nvidia GPUs).

Getting started: your first Cornac experiment

Flow of an Experiment in Cornac

Load the built-in MovieLens 100K dataset (will be downloaded if not cached):

import cornacml_100k = cornac.datasets.movielens.load_feedback(variant="100K")

Split the data based on ratio:

rs = cornac.eval_methods.RatioSplit(data=ml_100k, test_size=0.2, rating_threshold=4.0, seed=123)

Here we are comparing Biased MF, PMF, and BPR:

mf = cornac.models.MF(k=10, max_iter=25, learning_rate=0.01, lambda_reg=0.02, use_bias=True, seed=123)pmf = cornac.models.PMF(k=10, max_iter=100, learning_rate=0.001, lambda_reg=0.001, seed=123)bpr = cornac.models.BPR(k=10, max_iter=200, learning_rate=0.001, lambda_reg=0.01, seed=123)

Define metrics used to evaluate the models:

mae = cornac.metrics.MAE()rmse = cornac.metrics.RMSE()prec = cornac.metrics.Precision(k=10)recall = cornac.metrics.Recall(k=10)ndcg = cornac.metrics.NDCG(k=10)auc = cornac.metrics.AUC()mAP = cornac.metrics.MAP()

Put everything together into an experiment and run it:

cornac.Experiment( eval_method=rs, models=[mf, pmf, bpr], metrics=[mae, rmse, recall, ndcg, auc, mAP], user_based=True).run()

Output:

For more details, please take a look at our examples as well as tutorials.

Models

The recommender models supported by Cornac are listed below. Why don't you join us to lengthen the list?

Support

Your contributions at any level of the library are welcome. If you intend to contribute, please:

Fork the Cornac repository to your own account.Make changes and create pull requests.

You can also post bug reports and feature requests in GitHub issues.

Citation

If you use Cornac in a scientific publication, we would appreciate citations to the following paper:

Cornac: A Comparative Framework for Multimodal Recommender Systems, Salah et al., JMLR 21, pp. 1-5, 2020.

Bibtex entry:

@article{cornac, author = {Aghiles Salah and Quoc-Tuan Truong and Hady W. Lauw}, title = {Cornac: A Comparative Framework for Multimodal Recommender Systems}, journal = {Journal of Machine Learning Research}, year = {2020}, volume = {21}, number = {95}, pages = {1-5}, url = {http://jmlr.org/papers/v21/19-805.html}}

License

Apache License 2.0

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。