Spaces:
Sleeping
Sleeping
Added new model
Browse files- model_evals.json +61 -0
model_evals.json
CHANGED
@@ -181,5 +181,66 @@
|
|
181 |
"cognitively_normal": 0.9825072886297376
|
182 |
}
|
183 |
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
184 |
}
|
185 |
]
|
|
|
181 |
"cognitively_normal": 0.9825072886297376
|
182 |
}
|
183 |
}
|
184 |
+
},
|
185 |
+
{
|
186 |
+
"model_name": "/home/ubuntu/brain2vec",
|
187 |
+
"model_hf_path": "https://huggingface.co/radiata-ai/brain2vec-v2",
|
188 |
+
"model_nickname": "brain2vec-v2",
|
189 |
+
"model_type": "embedding",
|
190 |
+
"checkpoint": "/home/ubuntu/brain2vec/ae_output/autoencoder-ep-99.pth",
|
191 |
+
"checkpoint_hf_path": "https://huggingface.co/radiata-ai/brain2vec-v2/blob/main/autoencoder_final.pth",
|
192 |
+
"public": true,
|
193 |
+
"timestamp": 1741371749,
|
194 |
+
"train_l1": 0.014170298797539536,
|
195 |
+
"train_perceptual": 0.045079124881840384,
|
196 |
+
"train_ssim": 0.971753567151326,
|
197 |
+
"train_psnr": 30.71103699423002,
|
198 |
+
"val_l1": 0.03191869446987307,
|
199 |
+
"val_perceptual": 0.08965502869706232,
|
200 |
+
"val_ssim": 0.8541673189265828,
|
201 |
+
"val_psnr": 23.531947739459365,
|
202 |
+
"test_l1": 0.03180049764309693,
|
203 |
+
"test_perceptual": 0.08846248109098319,
|
204 |
+
"test_ssim": 0.8560211800379901,
|
205 |
+
"test_psnr": 23.602247131605857,
|
206 |
+
"downstream_train": {
|
207 |
+
"mae_age": 2.5919229984283447,
|
208 |
+
"acc_sex": 0.9971900245872849,
|
209 |
+
"sex_per_class_acc": {
|
210 |
+
"female": 0.9976798143851509,
|
211 |
+
"male": 0.9964381121994658
|
212 |
+
},
|
213 |
+
"acc_diag": 0.9983692106979778,
|
214 |
+
"diag_per_class_acc": {
|
215 |
+
"Alzheimers_disease": 0.9863013698630136,
|
216 |
+
"cognitively_normal": 0.9992975061468212
|
217 |
+
}
|
218 |
+
},
|
219 |
+
"downstream_val": {
|
220 |
+
"mae_age": 6.578178405761719,
|
221 |
+
"acc_sex": 0.8584070796460177,
|
222 |
+
"sex_per_class_acc": {
|
223 |
+
"female": 0.8803827751196173,
|
224 |
+
"male": 0.823076923076923
|
225 |
+
},
|
226 |
+
"acc_diag": 0.9615384615384616,
|
227 |
+
"diag_per_class_acc": {
|
228 |
+
"Alzheimers_disease": 0.64,
|
229 |
+
"cognitively_normal": 0.9852507374631269
|
230 |
+
}
|
231 |
+
},
|
232 |
+
"downstream_test": {
|
233 |
+
"mae_age": 6.33052921295166,
|
234 |
+
"acc_sex": 0.8425655976676385,
|
235 |
+
"sex_per_class_acc": {
|
236 |
+
"female": 0.8592964824120602,
|
237 |
+
"male": 0.8194444444444444
|
238 |
+
},
|
239 |
+
"acc_diag": 0.9697802197802198,
|
240 |
+
"diag_per_class_acc": {
|
241 |
+
"Alzheimers_disease": 0.8571428571428571,
|
242 |
+
"cognitively_normal": 0.9766763848396501
|
243 |
+
}
|
244 |
+
}
|
245 |
}
|
246 |
]
|