Federated learning - Githubissues

clinfo / kGCN

A graph-based deep learning framework for life science

Other

115 stars 35 forks source link

Federated learning #35

Closed ytkn closed 3 years ago

ytkn commented 3 years ago

Implemented a code to prepare toxicity dataset (toxicity.py). Running a training is not the purpose of this pull request, but I also implemented run_toxicity.py to confirm the dataset is correctly prepared.

run federated learning
```
python run_toxicity.py --federated
```
run training without federated learning
```
python run_toxicity.py
```

0h-n0 commented 3 years ago

Thank you for a pull request. It is no problems that this codes are merged into this branch.