Commit
·
263e38a
1
Parent(s):
734ed2d
Add evaluation results and output files to model card
Browse files- README.md +36 -0
- checkpoint-1264/config.json +22 -0
- checkpoint-1264/model.safetensors +3 -0
- checkpoint-1264/optimizer.pt +3 -0
- checkpoint-1264/preprocessor_config.json +23 -0
- checkpoint-1264/rng_state.pth +3 -0
- checkpoint-1264/scheduler.pt +3 -0
- checkpoint-1264/trainer_state.json +0 -0
- checkpoint-1264/training_args.bin +3 -0
- config.json +22 -0
- model.safetensors +3 -0
- pr_curve.png +0 -0
- preprocessor_config.json +23 -0
- roc_curve.png +0 -0
- training_args.bin +3 -0
README.md
ADDED
@@ -0,0 +1,36 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
language: code
|
3 |
+
tags:
|
4 |
+
- binary-classification
|
5 |
+
- model
|
6 |
+
- evaluation
|
7 |
+
metrics:
|
8 |
+
- average_precision: 0.93
|
9 |
+
- roc_auc: 0.90
|
10 |
+
- best threshold according to F1: 0.37
|
11 |
+
---
|
12 |
+
|
13 |
+
# Binary Classification Model
|
14 |
+
|
15 |
+
## Evaluation Results
|
16 |
+
|
17 |
+
**Average Precision:** 0.93
|
18 |
+
**ROC AUC:** 0.90
|
19 |
+
**best threshold according to F1: 0.37
|
20 |
+
|
21 |
+
|
22 |
+
## Visualizations
|
23 |
+
|
24 |
+
### Precision-Recall Curve
|
25 |
+

|
26 |
+
|
27 |
+
### ROC Curve
|
28 |
+

|
29 |
+
|
30 |
+
## Output Files and Directories
|
31 |
+
|
32 |
+
- 📂 `checkpoint-171/`
|
33 |
+
- `config.json`
|
34 |
+
- `model.safetensors`
|
35 |
+
- `preprocessor_config.json`
|
36 |
+
- `training_args.bin`
|
checkpoint-1264/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": 64,
|
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.48.0"
|
22 |
+
}
|
checkpoint-1264/model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:13c1198384fd0099bca87e9fd6a8b7a3544ab138680ccd8588b02e5555f8a799
|
3 |
+
size 8760128
|
checkpoint-1264/optimizer.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:fbdae7165cd7ad1ea2ba6de9f4acace8e2f38a4aaaae26335ccd72e908772a19
|
3 |
+
size 17564858
|
checkpoint-1264/preprocessor_config.json
ADDED
@@ -0,0 +1,23 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"do_convert_rgb": null,
|
3 |
+
"do_normalize": true,
|
4 |
+
"do_rescale": true,
|
5 |
+
"do_resize": false,
|
6 |
+
"image_mean": [
|
7 |
+
0.5,
|
8 |
+
0.5,
|
9 |
+
0.5
|
10 |
+
],
|
11 |
+
"image_processor_type": "ViTFeatureExtractor",
|
12 |
+
"image_std": [
|
13 |
+
0.5,
|
14 |
+
0.5,
|
15 |
+
0.5
|
16 |
+
],
|
17 |
+
"resample": 2,
|
18 |
+
"rescale_factor": 0.00392156862745098,
|
19 |
+
"size": {
|
20 |
+
"height": 32,
|
21 |
+
"width": 32
|
22 |
+
}
|
23 |
+
}
|
checkpoint-1264/rng_state.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9cb59f43a498ce8c7617e194a16d5255c4d94ecf089a151dd009526a8fc8016b
|
3 |
+
size 14244
|
checkpoint-1264/scheduler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d6aed05e4aebe899a19b3a80aedbdf89854334ea7eb1caf883b35635c8bebac1
|
3 |
+
size 1064
|
checkpoint-1264/trainer_state.json
ADDED
The diff for this file is too large to render.
See raw diff
|
|
checkpoint-1264/training_args.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d23bd679f2ac4710cc47143fd0d3ede7ba63efa5f001551630286ba32462adec
|
3 |
+
size 5432
|
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": 64,
|
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.48.0"
|
22 |
+
}
|
model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:19a6f54899e78fbef1b7ba0c10d4ae45fb33aeb499d950a1c219df118f36cbf8
|
3 |
+
size 8760128
|
pr_curve.png
ADDED
![]() |
preprocessor_config.json
ADDED
@@ -0,0 +1,23 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"do_convert_rgb": null,
|
3 |
+
"do_normalize": true,
|
4 |
+
"do_rescale": true,
|
5 |
+
"do_resize": false,
|
6 |
+
"image_mean": [
|
7 |
+
0.5,
|
8 |
+
0.5,
|
9 |
+
0.5
|
10 |
+
],
|
11 |
+
"image_processor_type": "ViTFeatureExtractor",
|
12 |
+
"image_std": [
|
13 |
+
0.5,
|
14 |
+
0.5,
|
15 |
+
0.5
|
16 |
+
],
|
17 |
+
"resample": 2,
|
18 |
+
"rescale_factor": 0.00392156862745098,
|
19 |
+
"size": {
|
20 |
+
"height": 32,
|
21 |
+
"width": 32
|
22 |
+
}
|
23 |
+
}
|
roc_curve.png
ADDED
![]() |
training_args.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d23bd679f2ac4710cc47143fd0d3ede7ba63efa5f001551630286ba32462adec
|
3 |
+
size 5432
|