Skip to content

Large Language Model for Generative Recommendation

License

Notifications You must be signed in to change notification settings

Alice-xyz/GenRec

 
 

Repository files navigation

GenRec: Large Language Model for Generative Recommendation

Install dependencies

pip install -r requirements.txt

Training (rec.py)

python rec.py \
    --base_model 'decapoda-research/llama-7b-hf' \
    --data_path './moives' \
    --output_dir './checkpoint'

Inference (generate.py)

python generate.py \
    --load_8bit \
    --base_model 'decapoda-research/llama-7b-hf' \
    --lora_weights './checkpoint/movies'

This project is implemented based on alpaca-lora (https://github.com/tloen/alpaca-lora)

Citation

@article{ji2023genrec,
  title={GenRec: Large Language Model for Generative Recommendation},
  author={Jianchao Ji, Zelong Li, Shuyuan Xu, Wenyue Hua, Yingqiang Ge, Juntao Tan and Yongfeng Zhang},
  journal={arXiv:2307.00457},
  year={2023}
}

About

Large Language Model for Generative Recommendation

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%