RAGTrans

A PyTorch implementation of RAGTrans(Retrieval-Augmented Hypergraph for Multimodal Social Media Popularity Prediction KDD 2024).

Dependencies

Install the dependencies via Anaconda:

• Python (>=3.8)
• PyTorch (>=2.1)
• NumPy (>=1.24)
• Scipy (>=1.11)
• torch-geometric(>=2.4)

create virtual environment:

conda create --name RAGTrans python=3.8

activate environment:

conda activate RAGTrans

install pytorch from pytorch:

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

To install all dependencies:

pip install -r requirements.txt