Skip to content

doheelab/lightfm_experiment

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

movielense 모델

tags: 테그 (from genome) ids: 제목

LightFM (tags)

  1. score : 0.663066
  2. user_ids (69878)
  3. item_ids (10681)
  4. item_features_matrix (10681, 1010)
  5. interactions (69878, 10681)

LightFM (tags + ids)

  1. score : 0.699208
  2. user_ids (69878)
  3. item_ids (10681)
  4. item_features_matrix (10681, 11689)
  5. interactions (69878, 10681)

Metadata Embeddings for User and Item Cold-start Recommendations

Pre-print available on arXiv.

Structure

  1. The paper directory holds the LaTeX source of the paper.
  2. The experiments directory holds the code used for experiments.

Reproducing results

To reproduce the results from the paper, do the following.

  1. Download and install the LightFM package (pip install lightfm).
  2. Download experiment data by running cd experiments/stackoverflow/ && make and cd experiments/movielens/ && make.
  3. To run the primary experiment, run ipython -- runner.py --table. This will print LaTeX variable definitions that, when pasted into the paper source file, populate the main results table.
  4. To generate the latent dimension sensitivity plots, run ipython -- runner.py --dim --plot. This will take some time.

It is possible to run invididual experiments. Examples:

  • ipython -- experiments/stackexchange/model.py --dim 50 --lsi --tags --ids --split 0.2 will run the CrossValidated experiment with 50-dimensional latent space, using the LSI-LR model with both post tags and post ids.
  • ipython -- experiments/movielens/model.py --ids --cold --split 0.2 will run the MF model on the MovieLens dataset in the cold-start scenario.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published