vamossyd commited on
Commit
7944dc0
β€’
1 Parent(s): b493a19

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +31 -0
README.md ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language:
3
+ - en
4
+ tags:
5
+ - text-classification
6
+ - emotion
7
+ - pytorch
8
+ license: mit
9
+ datasets:
10
+ - emotion
11
+ metrics:
12
+ - accuracy
13
+ - precision
14
+ - recall
15
+ - f1
16
+ ---
17
+
18
+ # bert-base-uncased-emotion
19
+
20
+ ## Model description
21
+
22
+ `bert-base-uncased` finetuned on the unify-emotion-datasets (https://github.com/sarnthil/unify-emotion-datasets), then transferred to
23
+ a small sample of 10K hand-tagged StockTwits messages. Optimized for extracting emotions from financial contexts.
24
+
25
+ Sequence length 64, learning rate 2e-5, batch size 128, 8 epochs.
26
+
27
+ For more details, please visit https://github.com/dvamossy/EmTract.
28
+
29
+ ## Training data
30
+
31
+ Data came from https://github.com/sarnthil/unify-emotion-datasets.