wuch15 / FedPerGNN

code of FedGNN

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

FedGNN

  1. Environment Requirements
  • Ubuntu 16.04
  • Anaconda with Python 3.6.9
  • CUDA 10.0

Note: The specific python package list of our environment is included in the requirements.txt. The tensorflow version can be 1.12~1.15. The installation may need several minutes if there is no environmental conflicts.

  1. Hardware requirements Needs a Linux server with larger than 32GB memory. GPU cores are optional but recommended.

  2. Running

  • Download datasets from their original sources
  • Convert then into matlab matrix formats (rows: users, columns: items, value: ratings)
  • Execute "python run.py"

Note: The logs will show the training loss and the test results. The estimated running time is from tens of minutes to hours, depending on the dataset size.

About

code of FedGNN

License:Apache License 2.0


Languages

Language:Python 100.0%