ZivKassnerNK commited on
Commit
5843710
·
1 Parent(s): 331a455

Add evaluation results and output files to model card

Browse files
README.md CHANGED
@@ -5,8 +5,8 @@ tags:
5
  - model
6
  - evaluation
7
  metrics:
8
- - average_precision: 0.94
9
- - roc_auc: 0.47
10
  - best threshold according to F1: 0.05
11
  ---
12
 
@@ -14,8 +14,8 @@ metrics:
14
 
15
  ## Evaluation Results
16
 
17
- **Average Precision:** 0.94
18
- **ROC AUC:** 0.47
19
  **best threshold according to F1: 0.05
20
 
21
 
 
5
  - model
6
  - evaluation
7
  metrics:
8
+ - average_precision: 0.97
9
+ - roc_auc: 0.75
10
  - best threshold according to F1: 0.05
11
  ---
12
 
 
14
 
15
  ## Evaluation Results
16
 
17
+ **Average Precision:** 0.97
18
+ **ROC AUC:** 0.75
19
  **best threshold according to F1: 0.05
20
 
21
 
checkpoint-1000/config.json ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "ViTForImageClassification"
4
+ ],
5
+ "attention_probs_dropout_prob": 0.0,
6
+ "encoder_stride": 16,
7
+ "hidden_act": "gelu",
8
+ "hidden_dropout_prob": 0.0,
9
+ "hidden_size": 256,
10
+ "image_size": 256,
11
+ "initializer_range": 0.02,
12
+ "intermediate_size": 512,
13
+ "layer_norm_eps": 1e-12,
14
+ "model_type": "vit",
15
+ "num_attention_heads": 8,
16
+ "num_channels": 3,
17
+ "num_hidden_layers": 4,
18
+ "patch_size": 16,
19
+ "qkv_bias": true,
20
+ "torch_dtype": "float32",
21
+ "transformers_version": "4.46.2"
22
+ }
checkpoint-1000/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8c57080a985a116d41d08ceac1980aa13ec799ee9e5db5c590eb5c36efc659b4
3
+ size 9497424
checkpoint-1000/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4b49262cdcc0768a20e310f14222b9105a6efae868dc4b408e6d35961b0044f2
3
+ size 19039290
checkpoint-1000/preprocessor_config.json ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_normalize": true,
3
+ "do_rescale": true,
4
+ "do_resize": false,
5
+ "image_mean": [
6
+ 0.5,
7
+ 0.5,
8
+ 0.5
9
+ ],
10
+ "image_processor_type": "ViTFeatureExtractor",
11
+ "image_std": [
12
+ 0.5,
13
+ 0.5,
14
+ 0.5
15
+ ],
16
+ "resample": 2,
17
+ "rescale_factor": 0.00392156862745098,
18
+ "size": {
19
+ "height": 256,
20
+ "width": 256
21
+ }
22
+ }
checkpoint-1000/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a467ee29ea3a4754f488af0c109338cd075b44bed51a7a1ac6fa1e89b9651e4b
3
+ size 14244
checkpoint-1000/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0f39d1cb1cb9393d208dbb9464c599bcb6392ca009bea38c9dbac28398b14819
3
+ size 1000
checkpoint-1000/trainer_state.json ADDED
The diff for this file is too large to render. See raw diff
 
checkpoint-1000/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:47936b623b1310e04416320d3a25c1d6b6643ea3be3af7f30d57f12bb9b17969
3
+ size 5368
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ded925128b7cb7aeb661a84c853a1cbc8cd32841405ddd44cf954cbdfc73d7d5
3
  size 9497424
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8c57080a985a116d41d08ceac1980aa13ec799ee9e5db5c590eb5c36efc659b4
3
  size 9497424
pr_curve.png CHANGED
roc_curve.png CHANGED
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e8762816cc9821828d767db1f935e5bda5bffeff8ebf233a280cc357a35f3e6e
3
  size 5368
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:47936b623b1310e04416320d3a25c1d6b6643ea3be3af7f30d57f12bb9b17969
3
  size 5368