Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

Sign up
Appearance settings

WHUIR/LLMRec

Repository files navigation

Quick Start:

  1. Download amazon dataset meta data(.json.gz) and rating data(.csv) files to dataset/Metadata and dataset/Ratings respectively.
  2. python preprocess_amazon.py
  3. python run_llmrec.py

Note:

  1. The code is based on Torch's DDP (Distributed Data Parallel) and defaults to using 4 GPUs for training.
  2. The code set the backbone to bert-base-uncased in default
  3. This code defaults to directly accessing the Hugging Face website to download pre-trained models.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

AltStyle によって変換されたページ (->オリジナル) /