mgyigit commited on
Commit
99ead72
·
verified ·
1 Parent(s): b2a0931

Update src/about.py

Browse files
Files changed (1) hide show
  1. src/about.py +5 -5
src/about.py CHANGED
@@ -124,16 +124,16 @@ If you find PROBE useful, please consider citing our work:
124
  Unsal, S., Atas, H., Albayrak, M., Turhan, K., Acar, A. C., & Doğan, T. (2022). Learning functional properties of proteins with language models. *Nature Machine Intelligence, 4*(3), 227-245.
125
  """
126
 
127
- TASK_INFO = ["Semantic Similarity", "GO-Based Function Prediction", "Target Protein Family Prediction", "Protein Protein Binding Affinity"]
128
  similarity_tasks_options = ["Sparse Uniform", "Well Annotated 200", "Well Annotated 500"]
129
  function_prediction_aspect_options = ["Molecular Function", "Biological Process", "Celular Components", "All Aspects"]
130
  family_prediction_dataset_options = ["Not-clustered", "UniClust50", "UniClust30", "MM-seqs15"]
131
 
132
  benchmark_type_map = {
133
- "Semantic Similarity": "similarity",
134
- "GO-Based Function Prediction": "function",
135
- "Target Protein Family Prediction": "family",
136
- "Protein Protein Binding Affinity": "affinity",
137
  }
138
 
139
  similarity_tasks_map = {
 
124
  Unsal, S., Atas, H., Albayrak, M., Turhan, K., Acar, A. C., & Doğan, T. (2022). Learning functional properties of proteins with language models. *Nature Machine Intelligence, 4*(3), 227-245.
125
  """
126
 
127
+ TASK_INFO = ["Semantic Similarity Inference", "Ontology-based Protein Function Prediction", "Drug Target Protein Family Classification", "Protein-Protein Binding Affinity Estimation"]
128
  similarity_tasks_options = ["Sparse Uniform", "Well Annotated 200", "Well Annotated 500"]
129
  function_prediction_aspect_options = ["Molecular Function", "Biological Process", "Celular Components", "All Aspects"]
130
  family_prediction_dataset_options = ["Not-clustered", "UniClust50", "UniClust30", "MM-seqs15"]
131
 
132
  benchmark_type_map = {
133
+ "Semantic Similarity Inference": "similarity",
134
+ "Ontology-based Function Prediction": "function",
135
+ "Drug Target Protein Family Classification": "family",
136
+ "Protein Protein Binding Affinity Estimation": "affinity",
137
  }
138
 
139
  similarity_tasks_map = {