mgyigit commited on
Commit
20d133f
·
verified ·
1 Parent(s): 0b73795

Update src/about.py

Browse files
Files changed (1) hide show
  1. src/about.py +9 -3
src/about.py CHANGED
@@ -100,9 +100,15 @@ family_prediction_dataset_options = ["nc", "uc50", "uc30", "mm15"]
100
 
101
  benchmark_specific_metrics = {
102
  'similarity': ['similarity_corr_MF', 'similarity_corr_BP', 'similarity_corr_CC', 'similarity_corr_Ave'],
103
- 'Benchmark 2': ['Metric B1', 'Metric B2', 'Metric B3'],
104
- 'Benchmark 3': ['Metric C1', 'Metric C2', 'Metric C3'],
105
- 'affinity': ['affinity_mse_train', 'affinity_mse_val']
 
 
 
 
 
 
106
  }
107
 
108
  display_labels = [
 
100
 
101
  benchmark_specific_metrics = {
102
  'similarity': ['similarity_corr_MF', 'similarity_corr_BP', 'similarity_corr_CC', 'similarity_corr_Ave'],
103
+ 'function': {
104
+ 'aspect_types': ['MF', 'BP', 'CC'],
105
+ 'dataset_types': ['accuracy', 'F1', 'precision', 'recall']
106
+ }
107
+ 'family': {
108
+ 'datasets': ['nc', 'uc30', 'uc50', 'mm15'],
109
+ 'metrics': ['F1', 'accuracy', 'mcc']
110
+ } 'Benchmark 3': ['Metric C1', 'Metric C2', 'Metric C3'],
111
+ 'affinity': ['mse', 'mae', 'corr']
112
  }
113
 
114
  display_labels = [