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!
sadaszewski1
/
decagon-pytorch
forked from sadaszewski/decagon-pytorch
1
0
Fork 0
Code Issues 0 Pull Requests 0 Releases 0 Wiki Activity
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.
205 Commits
1 Branch
866KB
Tree: 604b81675e
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '604b81675e'
${ noResults }
decagon-pytorch/tests/decagon_pytorch/test_data_matrix.py

14 lines
443B
Raw Blame History 

  1. from decagon_pytorch.data import Data

  2. 

  3. 

  4. def test_data():

  5.     d = Data()

  6.     d.add_node_type('Gene', 1000)

  7.     d.add_node_type('Drug', 100)

  8.     d.add_relation_type('Target', 1, 0, None)

  9.     d.add_relation_type('Interaction', 0, 0, None)

  10.     d.add_relation_type('Side Effect: Nausea', 1, 1, None)

  11.     d.add_relation_type('Side Effect: Infertility', 1, 1, None)

  12.     d.add_relation_type('Side Effect: Death', 1, 1, None)

  13.     print(d)