Sharpaxis commited on
Commit
c00f54d
·
verified ·
1 Parent(s): dae65d3
README.md ADDED
@@ -0,0 +1,78 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: apache-2.0
4
+ base_model: distilbert-base-uncased
5
+ tags:
6
+ - generated_from_trainer
7
+ datasets:
8
+ - financial_phrasebank
9
+ metrics:
10
+ - f1
11
+ model-index:
12
+ - name: Finance_DistilBERT_sentiment
13
+ results:
14
+ - task:
15
+ name: Text Classification
16
+ type: text-classification
17
+ dataset:
18
+ name: financial_phrasebank
19
+ type: financial_phrasebank
20
+ config: sentences_75agree
21
+ split: train
22
+ args: sentences_75agree
23
+ metrics:
24
+ - name: F1
25
+ type: f1
26
+ value: 0.9101001493367561
27
+ ---
28
+
29
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
30
+ should probably proofread and complete it, then remove this comment. -->
31
+
32
+ # Finance_DistilBERT_sentiment
33
+
34
+ This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the financial_phrasebank dataset.
35
+ It achieves the following results on the evaluation set:
36
+ - Loss: 0.2763
37
+ - F1: 0.9101
38
+ - Acc: 0.9088
39
+
40
+ ## Model description
41
+
42
+ More information needed
43
+
44
+ ## Intended uses & limitations
45
+
46
+ More information needed
47
+
48
+ ## Training and evaluation data
49
+
50
+ More information needed
51
+
52
+ ## Training procedure
53
+
54
+ ### Training hyperparameters
55
+
56
+ The following hyperparameters were used during training:
57
+ - learning_rate: 4e-06
58
+ - train_batch_size: 32
59
+ - eval_batch_size: 32
60
+ - seed: 42
61
+ - optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
62
+ - lr_scheduler_type: linear
63
+ - lr_scheduler_warmup_steps: 600
64
+ - num_epochs: 1
65
+
66
+ ### Training results
67
+
68
+ | Training Loss | Epoch | Step | Validation Loss | F1 | Acc |
69
+ |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|
70
+ | 0.0975 | 1.0 | 87 | 0.2763 | 0.9101 | 0.9088 |
71
+
72
+
73
+ ### Framework versions
74
+
75
+ - Transformers 4.46.2
76
+ - Pytorch 2.5.1
77
+ - Datasets 3.1.0
78
+ - Tokenizers 0.20.3
config.json ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "distilbert-base-uncased",
3
+ "activation": "gelu",
4
+ "architectures": [
5
+ "DistilBertForSequenceClassification"
6
+ ],
7
+ "attention_dropout": 0.1,
8
+ "dim": 768,
9
+ "dropout": 0.1,
10
+ "hidden_dim": 3072,
11
+ "id2label": {
12
+ "0": "LABEL_0",
13
+ "1": "LABEL_1",
14
+ "2": "LABEL_2"
15
+ },
16
+ "initializer_range": 0.02,
17
+ "label2id": {
18
+ "LABEL_0": 0,
19
+ "LABEL_1": 1,
20
+ "LABEL_2": 2
21
+ },
22
+ "max_position_embeddings": 512,
23
+ "model_type": "distilbert",
24
+ "n_heads": 12,
25
+ "n_layers": 6,
26
+ "pad_token_id": 0,
27
+ "problem_type": "single_label_classification",
28
+ "qa_dropout": 0.1,
29
+ "seq_classif_dropout": 0.2,
30
+ "sinusoidal_pos_embds": false,
31
+ "tie_weights_": true,
32
+ "torch_dtype": "float32",
33
+ "transformers_version": "4.46.2",
34
+ "vocab_size": 30522
35
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3200c0bde1eabb15eb3b5ee5c663e67ee2f1808d102378b78cb9875a87f62dfe
3
+ size 267835644
runs/Dec05_00-41-11_Aadityas-MacBook-Pro.local/events.out.tfevents.1733339473.Aadityas-MacBook-Pro.local.59022.11 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:971a2d39198db7d26603943ac5ccb221349534f20f0d0e159bc970b6f7c34527
3
+ size 12371
runs/Dec05_00-49-39_Aadityas-MacBook-Pro.local/events.out.tfevents.1733339983.Aadityas-MacBook-Pro.local.59022.12 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f8f0d799811e7fc17f4c1af7193eed1436d1a4e0cee900aacf71071cb603723c
3
+ size 6090
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ce6ff13bff9b18eb8155067e4d739fb50a30368fc947a52aff73c1899dca3f84
3
+ size 5304