gra1111/sentiment-bilstm-financial

Sentiment Bilstm Financial

Final SA BiLSTM head trained on top of the single-modal fine-tuned FinBERT (frozen). Input: raw tweet text.

Label mapping

LABEL_MAP = {"Bearish": 0, "Neutral": 1, "Bullish": 2}

Input format

tweet

Load

import torch
from huggingface_hub import hf_hub_download
from transformers import AutoModel, AutoTokenizer
from src.models import SentimentLSTM  # or copy class into your code

# 1. Download BiLSTM weights
weights_path = hf_hub_download(repo_id="gra1111/sentiment-bilstm-financial", filename="pytorch_model.pt")

# 2. Build the architecture using the config
config = {
  "input_dim": 768,
  "hidden_dim": 256,
  "num_layers": 3,
  "num_classes": 3,
  "dropout": 0.5,
  "input_format": "tweet",
  "finbert_repo": "gra1111/finbert-financial-sa"
}
model = SentimentLSTM(
    input_dim=config["input_dim"],
    hidden_dim=config["hidden_dim"],
    num_layers=config["num_layers"],
    num_classes=config["num_classes"],
    dropout=config["dropout"],
)
model.load_state_dict(torch.load(weights_path, map_location="cpu"))
model.eval()

# 3. Load FinBERT + tokenizer that match this BiLSTM
finbert = AutoModel.from_pretrained(config["finbert_repo"])
tokenizer = AutoTokenizer.from_pretrained(
    config.get("tokenizer_repo", config["finbert_repo"])
)