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!
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

test_data.py 407B

12345678910111213
  1. from decagon_pytorch.data import Data
  2. def test_data():
  3. d = Data()
  4. d.add_node_type('Gene')
  5. d.add_node_type('Drug')
  6. d.add_relation(1, 0, None, 'Target')
  7. d.add_relation(0, 0, None, 'Interaction')
  8. d.add_relation(1, 1, None, 'Side Effect: Nausea')
  9. d.add_relation(1, 1, None, 'Side Effect: Infertility')
  10. d.add_relation(1, 1, None, 'Side Effect: Death')
  11. print(d)