Skip to content
/ PAMCL Public
forked from nikke-rookie/PAMCL

The source code of "Preference-Aware Multimodal Contrastive Learning Recommendation Method" (PAMCL).

Notifications You must be signed in to change notification settings

sun2ot/PAMCL

 
 

Repository files navigation

Introduction

The source code of "Preference-Aware Multimodal Contrastive Learning Recommendation Method" (PAMCL).

Environment

The Python and package managers version are:

  • Python 3.9.19
  • mamba 1.5.8
  • conda 24.3.0

Use conda/mamba or pixi (Recommend) to install the environment:

conda env create -f environment.yml

pixi install

Datasets

Download from google drive.

Pre-trained models

Warning

Use bennegeek/stella_en_1.5B_v5 to generate text embeddings requires flash-attn which needs a CUDA version of at least 11.6, as indicated by nvcc -V.

Contributions

This project is forked from the original SELFRec. We would like to express our gratitude to Coder-Yu and his team for their outstanding work and open-source contributions.

The main changes are:

  • Support multi-modal datasets
  • Use Large Language Models to enhance presentations
  • Update some deprecated APIs
  • Improve code readability by type hint and docstrings
  • Refactor some modules

About

The source code of "Preference-Aware Multimodal Contrastive Learning Recommendation Method" (PAMCL).

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 58.9%
  • Python 41.1%