Code Monkey home page Code Monkey logo

caser_model's Introduction

Caser_Model

Overview : Tensorflow/Keras implementation of sequential recommendation system using convolutionally enhanced item embeddings and triplet loss.

For more information : https://arxiv.org/pdf/1809.07426.pdf

Data :

MovieLens 1M Dataset : https://grouplens.org/datasets/movielens/1m/
for simplicity dataset located in my github repo : https://raw.githubusercontent.com/malinphy/datasets/main/ml_1M/ratings.dat

File Description :

  • data_prep.py : generation of negative samples and target values
  • HelperFunctions.py : Data preparation for model training
  • model.py : Caser model written with tensoflow/keras
  • train.py : training file
  • caser_model_weights.h5 : model weights
  • eval.py : mean average precision MAP calculation
  • requirements.txt : required packages and versions to run model = caser_prediction.py : prediction file for deployment purpose

Usage : if necessary download repo and create an virtual env using following commands

download file

conda create --name caser_env
conda activate revenue_model

find the folder directory in caser_env

pip install -r requirements.txt 

run train.py file
for deployment purpose prediction file created seperately as caser_prediction.py

TODO :

original study investigated up to next 3 items. However, this implementation designed for 1 item. Designing will be developed according to original model.

Citation:

If you use this Caser in your paper, please cite the paper:

@inproceedings{tang2018caser,
  title={Personalized Top-N Sequential Recommendation via Convolutional Sequence Embedding},
  author={Tang, Jiaxi and Wang, Ke},
  booktitle={ACM International Conference on Web Search and Data Mining},
  year={2018}
}

caser_model's People

Contributors

malinphy avatar

Watchers

 avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.