IF YOU WOULD LIKE TO GET AN ACCOUNT, please write an email to s dot adaszewski at gmail dot com. User accounts are meant only to report issues and/or generate pull requests. This is a purpose-specific Git hosting for ADARED projects. Thank you for your understanding!
Stanislaw Adaszewski 3 лет назад
Родитель
Сommit
47645ef409
2 измененных файлов: 7 добавлений и 2 удалений
  1. +6
    -1
      README.md
  2. +1
    -1
      experiments/decagon_run/decagon_run.py

+ 6
- 1
README.md Просмотреть файл

@@ -9,9 +9,14 @@ settings.

Decagon-PyTorch is a PyTorch reimplementation of the algorithm.

## Citing

If you use this code in your research please cite this repository as:

Adaszewski S. (2020) https://code.adared.ch/sadaszewski/decagon-pytorch

## References

1. Zitnik, M., Agrawal, M., & Leskovec, J. (2018).
[Modeling polypharmacy side effects with graph convolutional networks](https://academic.oup.com/bioinformatics/article/34/13/i457/5045770)
Bioinformatics, 34(13), i457-i466.


+ 1
- 1
experiments/decagon_run/decagon_run.py Просмотреть файл

@@ -109,7 +109,7 @@ def _wrap(obj, method_name):
def main():
dev = torch.device('cuda:0')
dev = torch.device('cpu')
data = load_data(dev)
prep_d = prepare_training(data, TrainValTest(.8, .1, .1))
_wrap(Model, 'build')


Загрузка…
Отмена
Сохранить