|
|
@@ -10,7 +10,7 @@ def _some_data(): |
|
|
|
d.add_node_type('Gene', 1000)
|
|
|
|
d.add_node_type('Drug', 100)
|
|
|
|
|
|
|
|
fam = d.add_relation_family('Drug-Gene', 1, 0, False)
|
|
|
|
fam = d.add_relation_family('Drug-Gene', 1, 0, True)
|
|
|
|
fam.add_relation_type('Target', torch.rand(100, 1000))
|
|
|
|
|
|
|
|
fam = d.add_relation_family('Gene-Gene', 0, 0, False)
|
|
|
|