gtfintechlab/fomc_communication
Viewer • Updated • 2.48k • 371 • 10
How to use achen0525/DistilBERT_FOMC_Classifier with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="achen0525/DistilBERT_FOMC_Classifier") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("achen0525/DistilBERT_FOMC_Classifier")
model = AutoModelForSequenceClassification.from_pretrained("achen0525/DistilBERT_FOMC_Classifier")Fine-Tuned Transformer for FOMC Sentiment Classification
This model is a fine-tuned version of DistilBERT for FOMC meeting sentiment classification. It predicts whether a sentence from U.S. Federal Open Market Committee (FOMC) statements is Dovish, Hawkish, or Neutral.
from transformers import AutoTokenizer, AutoModelForSequenceClassification
import torch
model_name = "achen0525/DistilBERT_FOMC_Classifier"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForSequenceClassification.from_pretrained(model_name)
text = "The Committee decided to maintain the target range for the federal funds rate."
inputs = tokenizer(text, return_tensors="pt")
outputs = model(**inputs)
pred = torch.argmax(outputs.logits, dim=1)
labels = ['Dovish', 'Hawkish', 'Neutral']
print(labels[pred.item()])
For questions or feedback, reach out to: aochen@bu.edu
Base model
distilbert/distilbert-base-uncased