Skip to content

T
TFM_AdrianAyusoMuñoz
Commit c1efac0d authored by ADRIAN AYUSO MUNOZ's avatar ADRIAN AYUSO MUNOZ
Browse files
Options

Fix file

parent 6d263e1a
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 3 deletions
heterograph_construction.py
View file @ c1efac0d
...@@ -321,8 +321,7 @@ class DISNETConstructor: ...@@ -321,8 +321,7 @@ class DISNETConstructor:
# Add the edges to the graph. # Add the edges to the graph.
for edge_t in edges_dict.keys(): for edge_t in edges_dict.keys():
for edge in edges[ for edge in edges[edge_t]: # If a relation between a phenotype and a drug is in more than one type. The final type will be the last one.
edge_t]: # If a relation between a phenotype and a drug is in more than one type. The final type will be the last one.
try: try:
G.add_edge(int(edge[0]), int(edge[1]), edge_feature=edge[2], edge_type=edge_t) G.add_edge(int(edge[0]), int(edge[1]), edge_feature=edge[2], edge_type=edge_t)
except IndexError: except IndexError:
...@@ -404,7 +403,7 @@ class DISNETConstructor: ...@@ -404,7 +403,7 @@ class DISNETConstructor:
""" """
def decodePredictions(self, list, type, n=50, prepared=False, name=''): def decodePredictions(self, list, type, n=50, prepared=False, name=''):
dir = 'results/' + type + '_' + str(n) + name + '_' + 'table.csv' dir = 'results/' + type + '_' + str(n) + name + '_' + 'table.csv'
if (type == 'dis_dru_the'): if type == 'dis_dru_the':
# Read the data # Read the data
h, t = self.getNodeInfo(full=False) h, t = self.getNodeInfo(full=False)
# Give each node its ID # Give each node its ID
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment