andreadm commited on
Commit
fa5087d
·
verified ·
1 Parent(s): 28045aa

Add model card, licence, training config and per-seed evaluation

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -86,7 +86,7 @@ The fine-tuning and full-corpus inference code lives at
86
  | Pre-training | [`distilroberta-base`](https://huggingface.co/distilroberta-base) | 6-layer distilled RoBERTa, 82.1 M parameters |
87
  | Domain adaptation | [`mrm8488/distilroberta-finetuned-financial-news-sentiment-analysis`](https://huggingface.co/mrm8488/distilroberta-finetuned-financial-news-sentiment-analysis) | Financial-news sentiment |
88
  | Task adaptation | [`MAPAi/InflaBERT`](https://huggingface.co/MAPAi/InflaBERT) | Inflation-news sentiment (negative / neutral / positive) |
89
- | **This checkpoint** | **`andreadm/inflareddert`** | Full fine-tuning on Reddit titles, labels re-mapped to **down / neutral / up** |
90
 
91
  Architecture: `RobertaForSequenceClassification`, 6 hidden layers, hidden
92
  size 768, 12 attention heads, 50 265-token byte-level BPE vocabulary, 512-token
@@ -106,7 +106,7 @@ classification head was re-initialised for the three directional classes.
106
  ```python
107
  from transformers import pipeline
108
 
109
- clf = pipeline("text-classification", model="andreadm/inflareddert")
110
  clf("Fed officials warn prices will keep climbing as CPI hits 40-year high")
111
  # [{'label': 'up', 'score': 0.999}]
112
  ```
@@ -117,7 +117,7 @@ Batched, with the paper's `-1 / 0 / +1` encoding:
117
  import torch
118
  from transformers import AutoModelForSequenceClassification, AutoTokenizer
119
 
120
- name = "andreadm/inflareddert"
121
  tokenizer = AutoTokenizer.from_pretrained(name)
122
  model = AutoModelForSequenceClassification.from_pretrained(name).eval()
123
  encoding = {"down": -1, "neutral": 0, "up": 1}
 
86
  | Pre-training | [`distilroberta-base`](https://huggingface.co/distilroberta-base) | 6-layer distilled RoBERTa, 82.1 M parameters |
87
  | Domain adaptation | [`mrm8488/distilroberta-finetuned-financial-news-sentiment-analysis`](https://huggingface.co/mrm8488/distilroberta-finetuned-financial-news-sentiment-analysis) | Financial-news sentiment |
88
  | Task adaptation | [`MAPAi/InflaBERT`](https://huggingface.co/MAPAi/InflaBERT) | Inflation-news sentiment (negative / neutral / positive) |
89
+ | **This checkpoint** | **`andreadm/infla-reddit-bert`** | Full fine-tuning on Reddit titles, labels re-mapped to **down / neutral / up** |
90
 
91
  Architecture: `RobertaForSequenceClassification`, 6 hidden layers, hidden
92
  size 768, 12 attention heads, 50 265-token byte-level BPE vocabulary, 512-token
 
106
  ```python
107
  from transformers import pipeline
108
 
109
+ clf = pipeline("text-classification", model="andreadm/infla-reddit-bert")
110
  clf("Fed officials warn prices will keep climbing as CPI hits 40-year high")
111
  # [{'label': 'up', 'score': 0.999}]
112
  ```
 
117
  import torch
118
  from transformers import AutoModelForSequenceClassification, AutoTokenizer
119
 
120
+ name = "andreadm/infla-reddit-bert"
121
  tokenizer = AutoTokenizer.from_pretrained(name)
122
  model = AutoModelForSequenceClassification.from_pretrained(name).eval()
123
  encoding = {"down": -1, "neutral": 0, "up": 1}