grkon97 commited on
Commit
6a3bae1
·
verified ·
1 Parent(s): 393f8c0

Model save

Browse files
Files changed (2) hide show
  1. README.md +77 -0
  2. model.safetensors +1 -1
README.md ADDED
@@ -0,0 +1,77 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: cc-by-nc-4.0
4
+ base_model: MCG-NJU/videomae-base
5
+ tags:
6
+ - generated_from_trainer
7
+ metrics:
8
+ - accuracy
9
+ model-index:
10
+ - name: Intoxicated_Classification
11
+ results: []
12
+ ---
13
+
14
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
15
+ should probably proofread and complete it, then remove this comment. -->
16
+
17
+ # Intoxicated_Classification
18
+
19
+ This model is a fine-tuned version of [MCG-NJU/videomae-base](https://huggingface.co/MCG-NJU/videomae-base) on an unknown dataset.
20
+ It achieves the following results on the evaluation set:
21
+ - Loss: 1.7660
22
+ - Accuracy: 0.6971
23
+ - F1 Macro: 0.6962
24
+ - F1 Weighted: 0.6990
25
+ - Precision Macro: 0.7040
26
+ - Precision Weighted: 0.7216
27
+ - Recall Macro: 0.7094
28
+ - Recall Weighted: 0.6971
29
+
30
+ ## Model description
31
+
32
+ More information needed
33
+
34
+ ## Intended uses & limitations
35
+
36
+ More information needed
37
+
38
+ ## Training and evaluation data
39
+
40
+ More information needed
41
+
42
+ ## Training procedure
43
+
44
+ ### Training hyperparameters
45
+
46
+ The following hyperparameters were used during training:
47
+ - learning_rate: 5e-05
48
+ - train_batch_size: 8
49
+ - eval_batch_size: 8
50
+ - seed: 42
51
+ - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
52
+ - lr_scheduler_type: linear
53
+ - lr_scheduler_warmup_ratio: 0.1
54
+ - training_steps: 6980
55
+
56
+ ### Training results
57
+
58
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 Macro | F1 Weighted | Precision Macro | Precision Weighted | Recall Macro | Recall Weighted |
59
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|:--------:|:-----------:|:---------------:|:------------------:|:------------:|:---------------:|
60
+ | 0.3101 | 0.1 | 698 | 1.1498 | 0.6133 | 0.6127 | 0.6100 | 0.6448 | 0.6663 | 0.6396 | 0.6133 |
61
+ | 0.0223 | 1.1 | 1396 | 2.0621 | 0.6154 | 0.6114 | 0.6046 | 0.6727 | 0.6992 | 0.6528 | 0.6154 |
62
+ | 0.071 | 2.1 | 2094 | 1.1017 | 0.6650 | 0.6649 | 0.6641 | 0.6905 | 0.7124 | 0.6884 | 0.6650 |
63
+ | 0.1562 | 3.1 | 2792 | 0.9922 | 0.7803 | 0.7713 | 0.7792 | 0.7746 | 0.7791 | 0.7691 | 0.7803 |
64
+ | 0.1505 | 4.1 | 3490 | 0.6705 | 0.8203 | 0.8157 | 0.8208 | 0.8143 | 0.8217 | 0.8176 | 0.8203 |
65
+ | 0.0634 | 5.1 | 4188 | 2.0951 | 0.6155 | 0.6122 | 0.6059 | 0.6687 | 0.6945 | 0.6515 | 0.6155 |
66
+ | 0.0002 | 6.1 | 4886 | 2.2097 | 0.6473 | 0.6472 | 0.6483 | 0.6627 | 0.6818 | 0.6645 | 0.6473 |
67
+ | 0.0366 | 7.1 | 5584 | 1.7946 | 0.6842 | 0.6837 | 0.6858 | 0.6947 | 0.7133 | 0.6988 | 0.6842 |
68
+ | 0.0002 | 8.1 | 6282 | 1.8526 | 0.6838 | 0.6832 | 0.6856 | 0.6926 | 0.7107 | 0.6972 | 0.6838 |
69
+ | 0.0004 | 9.1 | 6980 | 1.7660 | 0.6971 | 0.6962 | 0.6990 | 0.7040 | 0.7216 | 0.7094 | 0.6971 |
70
+
71
+
72
+ ### Framework versions
73
+
74
+ - Transformers 4.51.3
75
+ - Pytorch 2.6.0+cu124
76
+ - Datasets 3.5.1
77
+ - Tokenizers 0.21.1
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4c55fb0d1861cfe6a7ee26b5d37f1ce9b8ac49517303f800f014fed27e250587
3
  size 344937368
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:05c4d07eddfb1a6eda0b441c593a2099533b13430f595f29164a263fe8ebe795
3
  size 344937368