Hugging Face's logo

Danieli1021
/
Resnet50_backend

Safetensors
custom-resnet50-binary-classifier
Model card Files
xet
 Community
Danieli1021 commited on
Commit
07fa5a7
·
verified ·
1 Parent(s): b600590

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +23 -23
config.json CHANGED
@@ -1,23 +1,23 @@
1
- {
2
- "architectures": [
3
- "RoadworkBinaryClassifier"
4
- ],
5
- "model_type": "custom-resnet50-binary-classifier",
6
- "backbone_name": "resnet50",
7
- "num_labels": 1,
8
- "id2label": {
9
- "0": "non-roadwork",
10
- "1": "roadwork"
11
- },
12
- "label2id": {
13
- "non-roadwork": 0,
14
- "roadwork": 1
15
- },
16
- "problem_type": "binary_classification",
17
- "classifier_activation": "sigmoid",
18
- "decision_threshold": 0.5,
19
- "image_size": 224,
20
- "torch_dtype": "float32",
21
- "best_val_f1": 0.9349030470914127,
22
- "best_val_validator_reward_proxy": 0.5447887084945165
23
- }
 
1
+ {
2
+ "architectures": [
3
+ "RoadworkBinaryClassifier"
4
+ ],
5
+ "model_type": "custom-resnet50-binary-classifier",
6
+ "backbone_name": "resnet50",
7
+ "num_labels": 1,
8
+ "id2label": {
9
+ "0": "None",
10
+ "1": "Roadwork"
11
+ },
12
+ "label2id": {
13
+ "None": 0,
14
+ "Roadwork": 1
15
+ },
16
+ "problem_type": "binary_classification",
17
+ "classifier_activation": "sigmoid",
18
+ "decision_threshold": 0.5,
19
+ "image_size": 224,
20
+ "torch_dtype": "float32",
21
+ "best_val_f1": 0.9349030470914127,
22
+ "best_val_validator_reward_proxy": 0.5447887084945165
23
+ }