Sparseness of user-to-item rating data is one of the major factors that deteriorate the quality of recommender system. To handle the sparsity problem, several recommendation techniques have been proposed that additionally consider auxiliary information to improve rating prediction accuracy. In particular, when rating data is sparse, document modeling-based approaches have improved the accuracy by additionally utilizing textual data such as reviews, abstracts, or synopses. However, due to the inherent limitation of the bag-of-words model, they have difficulties in effectively utilizing contextual information of the documents, which leads to shallow understanding of the documents. This paper proposes a novel context-aware recommendation model, convolutional matrix factorization (ConvMF) that integrates convolutional neural network (CNN) into probabilistic matrix factorization (PMF). Consequently, ConvMF captures contextual information of documents and further enhances the rating prediction accuracy.
- Convolutional Matrix Factorization for Document Context-Aware Recommendation (RecSys 2016)
- Donghyun Kim, Chanyoung Park, Jinoh Oh, Seungyong Lee, Hwanjo Yu
- Deep Hybrid Recommender Systems via Exploiting Document Context and Statistics of Items (Information Sciences (SCI))
- Donghyun Kim, Chanyoung Park, Jinoh Oh, Hwanjo Yu
- Python 2.7
- Keras 0.3.3
- See installation instructions
Note: Run python <install_path>/run.py -h
in bash shell. You will see how to configure several parameters for ConvMF
You can evaluate our model with different settings in terms of the size of dimension, the value of hyperparameter, the number of convolutional kernal, and etc. Below is a description of all the configurable parameters and their defaults:
Parameter | Default |
---|---|
-h , --help |
{} |
-c <bool> , --do_preprocess <bool> |
False |
-r <path> , --raw_rating_data_path <path> |
{} |
-i <path> , --raw_item_document_data_path <path> |
{} |
-m <integer> , --min_rating <integer> |
{} |
-l <integer> , --max_length_document <integer> |
300 |
-f <float> , --max_df <float> |
0.5 |
-s <integer> , --vocab_size <integer> |
8000 |
-t <float> , --split_ratio <float> |
0.2 |
-d <path> , --data_path <path> |
{} |
-a <path> , --aux_path <path> |
{} |
-o <path> , --res_dir <path> |
{} |
-e <integer> , --emb_dim <integer> |
200 |
-p <path> , --pretrain_w2v <path> |
{} |
-g <bool> , --give_item_weight <bool> |
True |
-k <integer> , --dimension <integer> |
50 |
-u <float> , --lambda_u <float> |
{} |
-v <float> , --lambda_v <float> |
{} |
-n <integer> , --max_iter <integer> |
200 |
-w <integer> , --num_kernel_per_ws |
100 |
do_preprocess
:True
orFalse
in order to preprocess raw data for ConvMF.raw_rating_data_path
: path to a raw rating data path. The data format should beuser id::item id::rating
.min_rating
: users who have less thanmin_rating
ratings will be removed.max_length_document
: the maximum length of document of each item.max_df
: threshold to ignore terms that have a document frequency higher than the given value. i.e. for removing corpus-stop words.vocab_size
: size of vocabulary.split_ratio
: 1-ratio, ratio/2 and ratio/2 of the entire dataset will be constructed as training, valid and test set, respectively.data_path
: path to training, valid and test datasets.aux_path
: path to R, D_all sets that are generated during the preprocessing step.res_dir
: path to ConvMF's resultemb_dim
: the size of latent dimension for word vectors.pretrain_w2v
: path to pretrained word embedding model to initialize word vectors.give_item_weight
:True
orFalse
to give item weight for R-ConvMF.dimension
: the size of latent dimension for users and items.lambda_u
: parameter of user regularizer.lambda_v
: parameter of item regularizer.max_iter
: the maximum number of iteration.num_kernel_per_ws
: the number of kernels per window size for CNN module.
python3 rec_eval/lib/split.py -d ${rootpath}/${dataFolder} -s in-matrix-item -v
Convert the citeulike data format
python3 process_citeulike.py --data_dir ../data --dataset citeulike-a
python run.py --do_preprocess true --raw_rating_data_path /vol2/wanliz/data/Extended_ctr/convmf/citeulike_a_extended/ratings.txt --raw_item_document_data_path /vol2/wanliz/data/Extended_ctr/convmf/citeulike_a_extended/papers.txt -d /vol2/wanliz/data/Extended_ctr/convmf/citeulike_a_extended/inmatrix -a /vol2/wanliz/data/Extended_ctr/convmf/citeulike_a_extended/preprocessed/ --splits_dir /vol2/wanliz/data/Extended_ctr/convmf/citeulike_a_extended/in-matrix-item_folds/fold-1 -o /vol2/wanliz/data/Extended_ctr/convmf/citeulike_a_extended/results/inmatrix/
python run.py -d /home/zaher/data/Extended_ctr/convmf/dummy/preprocessed/inmatrix/fold-1 -a /home/zaher/data/Extended_ctr/convmf/dummy/preprocessed/ -o /home/zaher/data/Extended_ctr/convmf/citeulike_a_extended/results/ -e 200 -k 200 -p /home/zaher/data/cbow_w2v/w2v_200.txt -u 10 -v 100