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.

14 line
443B

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