ash12321 commited on
Commit
7cf3d3c
·
verified ·
1 Parent(s): 5b411e0

Add model config

Browse files
Files changed (1) hide show
  1. config.json +9 -8
config.json CHANGED
@@ -1,12 +1,13 @@
1
  {
2
- "model_type": "resnet50",
3
- "task": "image-classification",
4
  "num_classes": 2,
5
- "class_names": [
6
- "real",
7
- "fake"
 
8
  ],
9
- "image_size": 256,
10
- "generator_model": "stable-diffusion-xl-base-1.0",
11
- "random_seed": 42
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
  }