Downloads · 30 days
5
18% of all-time downloads
otaviosbampato/finance-emotions-model
finance-emotions-model is a machine learning model from otaviosbampato. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
this is a simple project created as part of a learning experience on the intricate functioning of LLM's and transformers. it uses a BERT (encoder-only) model to infer emotions from small phrases in finance. emotions c…
Downloads · 30 days
5
18% of all-time downloads
All-time downloads
28
Public
Parameters
108M
433 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors433 MB · 100%
From the Hugging Face model README
this is a simple project created as part of a learning experience on the intricate functioning of LLM's and transformers. it uses a BERT (encoder-only) model to infer emotions from small phrases in finance. emotions can be classified as either one of seven labels: neutral, happy, fear, anger, sad, disgust, surprise.
uses BERT (encoder-only) for sequence classification.
tokenization with bert-base-cased.
custom mapping from emotion labels to numeric IDs.
training and evaluation with Hugging Face Trainer.
accuracy metric using evaluate.