Kroalist commited on
Commit
7b7ba46
verified
1 Parent(s): 41709ac

Add/refresh model card and checkpoint

Browse files
Files changed (3) hide show
  1. README.md +23 -0
  2. config.json +7 -7
  3. pytorch_model.bin +1 -1
README.md ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Financial Sentiment BERT (FinBERT fine-tune)
2
+
3
+ **Repo ID:** `Kroalist/financial-sentiment-bert`
4
+ **Base model:** [`ProsusAI/finbert`](https://huggingface.co/ProsusAI/finbert)
5
+ **Dataset:** Financial PhraseBank (Malo et al.)
6
+ **Classes:** 0 = positive 路 1 = negative 路 2 = neutral
7
+ **Fine-tuning epochs:** 6
8
+ **Optimizer:** AdamW (lr = 2e-5)
9
+ **Hardware:** CPU-only training
10
+
11
+ ## Intended use
12
+
13
+ Sentence-level sentiment tagging of English financial news, reports, or earnings calls to feed dashboards or downstream trading models.
14
+
15
+ ## Quick load
16
+
17
+ ```python
18
+ from transformers import AutoTokenizer, AutoModelForSequenceClassification
19
+ tok = AutoTokenizer.from_pretrained("Kroalist/financial-sentiment-bert")
20
+ model = AutoModelForSequenceClassification.from_pretrained("Kroalist/financial-sentiment-bert")
21
+ ```
22
+
23
+ _Last updated: 2025-04-23_
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "/Users/arthursavary/Desktop/Code_AI/final_project/best_model",
3
  "architectures": [
4
  "BertForSequenceClassification"
5
  ],
@@ -10,16 +10,16 @@
10
  "hidden_dropout_prob": 0.1,
11
  "hidden_size": 768,
12
  "id2label": {
13
- "0": "negative",
14
- "1": "neutral",
15
- "2": "positive"
16
  },
17
  "initializer_range": 0.02,
18
  "intermediate_size": 3072,
19
  "label2id": {
20
- "negative": 0,
21
- "neutral": 1,
22
- "positive": 2
23
  },
24
  "layer_norm_eps": 1e-12,
25
  "max_position_embeddings": 512,
 
1
  {
2
+ "_name_or_path": "ProsusAI/finbert",
3
  "architectures": [
4
  "BertForSequenceClassification"
5
  ],
 
10
  "hidden_dropout_prob": 0.1,
11
  "hidden_size": 768,
12
  "id2label": {
13
+ "0": "positive",
14
+ "1": "negative",
15
+ "2": "neutral"
16
  },
17
  "initializer_range": 0.02,
18
  "intermediate_size": 3072,
19
  "label2id": {
20
+ "negative": 1,
21
+ "neutral": 2,
22
+ "positive": 0
23
  },
24
  "layer_norm_eps": 1e-12,
25
  "max_position_embeddings": 512,
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:162dcd26e997f6a83a09b4d4b7867164c174e33bcdf165d357ce46c6c2d2208a
3
  size 438008181
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0fac89cc799e1376543aebda46dadf6b874cd592d9bf3e7cc084a14ec7c70d78
3
  size 438008181