Add model config
Browse files- config.json +9 -8
config.json
CHANGED
|
@@ -1,12 +1,13 @@
|
|
| 1 |
{
|
| 2 |
-
"model_type": "
|
| 3 |
-
"
|
| 4 |
"num_classes": 2,
|
| 5 |
-
"
|
| 6 |
-
|
| 7 |
-
"
|
|
|
|
| 8 |
],
|
| 9 |
-
"
|
| 10 |
-
"
|
| 11 |
-
"
|
| 12 |
}
|
|
|
|
| 1 |
{
|
| 2 |
+
"model_type": "image-classification",
|
| 3 |
+
"architecture": "resnet50",
|
| 4 |
"num_classes": 2,
|
| 5 |
+
"image_size": 224,
|
| 6 |
+
"labels": [
|
| 7 |
+
"Real",
|
| 8 |
+
"SDXL-generated"
|
| 9 |
],
|
| 10 |
+
"test_accuracy": 99.75490196078431,
|
| 11 |
+
"test_f1": 99.77250568735782,
|
| 12 |
+
"test_auc_roc": 0.9998845880681818
|
| 13 |
}
|