Spaces:
Sleeping
Sleeping
Update src/bin/PROBE.py
Browse files- 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"],
|
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):
|