ZivKassnerNK commited on
Commit
93f2955
·
1 Parent(s): 8f3f9f2

Add evaluation results and output files to model card

Browse files
README.md CHANGED
@@ -5,18 +5,18 @@ tags:
5
  - model
6
  - evaluation
7
  metrics:
8
- - average_precision: 0.76
9
- - roc_auc: 0.74
10
- - best threshold according to F1: 0.49
11
  ---
12
 
13
  # Binary Classification Model
14
 
15
  ## Evaluation Results
16
 
17
- **Average Precision:** 0.76
18
- **ROC AUC:** 0.74
19
- **best threshold according to F1: 0.49
20
 
21
 
22
  ## Visualizations
 
5
  - model
6
  - evaluation
7
  metrics:
8
+ - average_precision: 0.71
9
+ - roc_auc: 0.67
10
+ - best threshold according to F1: 0.06
11
  ---
12
 
13
  # Binary Classification Model
14
 
15
  ## Evaluation Results
16
 
17
+ **Average Precision:** 0.71
18
+ **ROC AUC:** 0.67
19
+ **best threshold according to F1: 0.06
20
 
21
 
22
  ## Visualizations
checkpoint-4903/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": 32,
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": 4,
19
+ "qkv_bias": true,
20
+ "torch_dtype": "float32",
21
+ "transformers_version": "4.46.1"
22
+ }
checkpoint-4903/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4c2567a4e58b552eb58c55687da100423da92883558224b436482222799a2315
3
+ size 8563512
checkpoint-4903/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ce9a7f2d3b7a3c25d44126d7db525891dda3c8e35f11fd03b52acab9565442d5
3
+ size 17171514
checkpoint-4903/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": 32,
20
+ "width": 32
21
+ }
22
+ }
checkpoint-4903/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8ee669d500ce1549dd54c16906eb9e37bb35535f0d0e4c04fddeee46afce1408
3
+ size 14244
checkpoint-4903/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7053e9494ca8bdb39c15c8a2d08ee749d716fa63ea9b8c209f58ee32c2c59a31
3
+ size 1000
checkpoint-4903/trainer_state.json ADDED
The diff for this file is too large to render. See raw diff
 
checkpoint-4903/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:63823c9b3160b4e4798bc0506bf9a94bc5b26c230d202e1d19b351619dd3852d
3
+ size 5368
config.json CHANGED
@@ -1,5 +1,4 @@
1
  {
2
- "_name_or_path": "GenetikaPlus/binary_classification_model_v3.1.5_Junction",
3
  "architectures": [
4
  "ViTForImageClassification"
5
  ],
 
1
  {
 
2
  "architectures": [
3
  "ViTForImageClassification"
4
  ],
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:52618ad8b197f9deaf5512201fbac498b07ccf0b9b0316dbc0c95c56993c967a
3
  size 8563512
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cf13c1caaa28733c7968808610e83fb5d6a7643734cbe5aeb0c25716866fda9e
3
  size 8563512
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:479cb5ccfd6e92e0488f7f0b7874ba9c23dcf91276c3b39bcff6a7a26c3f5254
3
  size 5368
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:63823c9b3160b4e4798bc0506bf9a94bc5b26c230d202e1d19b351619dd3852d
3
  size 5368