MCNB

Prerequisites:

python 3.8.8

CUDA 11.2

torch 1.9.0

torchvision 0.10.0

sklearn 0.24.1

numpy 1.20.1

scipy 1.6.2

pandas 1.2.4

• packages.py - all the needed packages
• load_data.py - load the datasets
• run.py - run the methods
• read_movie.ipynb - show how we processed the MovieLens data

Methods:

"locb", "club", "sclub", "cofiba", "neuucb_one", "neuucb_ind", "mcnb"

Datasets:

"movie"

Run:

python run.py --dataset "dataset" --method "method"