UFGraphFR: Graph Federation Recommendation System based on User Text description features
mkdir log mkdir sh_result mkdir res mkdir checkpoint
matplotlib==3.9.2
mediapipe==0.10.15
pandas==2.2.2
scikit-learn==1.5.1
sentence-transformers==3.2.1
tensorboard==2.17.1
tensorboardX==2.6.2.2
torch==2.6.0.dev20241017
torch_geometric==2.5.3
torchaudio==2.5.0.dev20241017
torchvision==0.20.0.dev20241017
tqdm
python train.py
If you find this project helpful, please consider to cite the following paper: