Who You Are Matters: Bridging Topics and Social Roles via LLM-Enhanced Logical Recommendation

Requirements

• Python 3.7+
• PyTorch 1.12+
• CUDA 11.6+

Installation

RecBole works with the following operating systems:

• Linux
• Windows 10
• macOS X

RecBole requires Python version 3.7 or later.

Install from conda

conda install -c aibox recbole

Install from pip

pip install recbole

Install from source

git clone https://github.com/RUCAIBox/RecBole.git && cd RecBole
pip install -e . --verbose

Run

With the source code, you can use the provided script for initial usage of our method:

python run_tagcf.py --dataset=amazon_movie

This script will run the LogicRec model on the Amazon Movie dataset.

If you want to change the parameters, such as learning_rate, embedding_size, just set the additional command parameters as you need:

python run_tagcf.py --learning_rate=0.0001 --embedding_size=128

If you want to change the models, just run the script by setting additional command parameters:

python run_tagcf.py --model=[model_name]