mgyigit commited on
Commit
39ebd1e
1 Parent(s): c806fef

Update src/bin/PROBE.py

Browse files
Files changed (1) hide show
  1. src/bin/PROBE.py +1 -1
src/bin/PROBE.py CHANGED
@@ -15,7 +15,7 @@ def load_representation(multi_col_representation_vector_file_path):
15
  original_values_as_df.loc[index] = [multi_col_representation_vector.iloc[index]['Entry']] + [list_of_floats]
16
  return original_values_as_df
17
 
18
- 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):
19
  print("\n\nPROBE (Protein RepresentatiOn Benchmark) run is started...\n\n")
20
 
21
  if any(item in ['similarity', 'function', 'family', 'all'] for item in benchmarks):
 
15
  original_values_as_df.loc[index] = [multi_col_representation_vector.iloc[index]['Entry']] + [list_of_floats]
16
  return original_values_as_df
17
 
18
+ def run_probe(benchmarks, representation_name, representation_file_human, representation_file_affinity, similarity_tasks=["Sparse","200","500"], function_prediction_aspect="All_Aspects", function_prediction_dataset="All_Data_Sets", family_prediction_dataset=["nc","uc50","uc30","mm15"], detailed_output=False):
19
  print("\n\nPROBE (Protein RepresentatiOn Benchmark) run is started...\n\n")
20
 
21
  if any(item in ['similarity', 'function', 'family', 'all'] for item in benchmarks):