Update src/bin/PROBE.py

src/bin/PROBE.py

@@ -17,14 +17,7 @@ def load_representation(multi_col_representation_vector_file_path):
 
 def run_probe(benchmarks, representation_name, representation_file_human, representation_file_affinity, similarity_tasks=["Sparse","200","500"], function_prediction_aspec="All_Aspects", function_prediction_dataset="All_Data_Sets", family_prediction_dataset=["nc","uc50","uc30","mm15"], detailed_output=False):
     print("\n\nPROBE (Protein RepresentatiOn Benchmark) run is started...\n\n")
-    print(type(benchmarks))
-    print(benchmarks)
-    if any(item in ['similarity', 'function', 'family', 'all'] for item in benchmarks):
-        print("AAAAAAAAA")
-    else:
-        print("BBBBBBBB")
     
-
     if any(item in ['similarity', 'function', 'family', 'all'] for item in benchmarks):
         print("\nRepresentation vectors are loading...\n")
         human_representation_dataframe = load_representation(representation_file_human)
@@ -36,7 +29,11 @@ def run_probe(benchmarks, representation_name, representation_file_human, repres
         ssi.protein_names = ssi.representation_dataframe['Entry'].tolist()
         ssi.similarity_tasks = similarity_tasks
         ssi.detailed_output = detailed_output
-        ssi.calculate_all_correlations()
+        print("IN SIMILARITY CALC")
+        similarity_result = ssi.calculate_all_correlations()
+        print("OUT SIMILARITY CALC")
+        print(similarity_result)
+        
 
     if "function" in benchmarks:
         print("\n\nOntology-based protein function prediction benchmark is running...\n")