annotations -> prop_annotations

bio-ontology-research-group · Jul 6, 2021 · 5f87260 · 5f87260
1 parent 1e84042
commit 5f87260
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/evaluate_diamondblast.py b/evaluate_diamondblast.py
@@ -48,12 +48,12 @@ def main(train_data_file, test_data_file, diamond_scores_file, ont):
     terms_dict = {v: i for i, v in enumerate(terms)}
 
     train_df = pd.read_pickle(train_data_file)
-    annotations = train_df['annotations'].values
+    annotations = train_df['prop_annotations'].values
     annotations = list(map(lambda x: set(x), annotations))
 
     test_df = pd.read_pickle(test_data_file)
     # Annotations
-    test_annotations = test_df['annotations'].values
+    test_annotations = test_df['prop_annotations'].values
     test_annotations = list(map(lambda x: set(x), test_annotations))
 
     go_rels.calculate_ic(annotations + test_annotations)

diff --git a/evaluate_diamondscore.py b/evaluate_diamondscore.py
@@ -45,11 +45,11 @@ def main(train_data_file, test_data_file, diamond_scores_file, ont):
     go_rels = Ontology('data/go.obo', with_rels=True)
 
     train_df = pd.read_pickle(train_data_file)
-    annotations = train_df['annotations'].values
+    annotations = train_df['prop_annotations'].values
     annotations = list(map(lambda x: set(x), annotations))
 
     test_df = pd.read_pickle(test_data_file)
-    test_annotations = test_df['annotations'].values
+    test_annotations = test_df['prop_annotations'].values
     test_annotations = list(map(lambda x: set(x), test_annotations))
     go_rels.calculate_ic(annotations + test_annotations)