Trading Hero LLM

🚀 Trading Hero Financial Sentiment Analysis

This model is a fine - tuned version of FinBERT, designed for financial sentiment analysis, which can enhance performance in specific financial NLP tasks.

🚀 Quick Start

This model is a fine - tuned version of FinBERT, a BERT model pre - trained on financial texts. The fine - tuning process was carried out to adapt the model to specific financial NLP tasks, improving its performance in domain - specific sentiment analysis applications.

✨ Features

• Target Users: Financial analysts, NLP researchers, and developers working on financial data.

📦 Installation

No specific installation steps are provided in the original document, so this section is skipped.

💻 Usage Examples

Basic Usage

import torch
from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
tokenizer = AutoTokenizer.from_pretrained("fuchenru/Trading-Hero-LLM")
model = AutoModelForSequenceClassification.from_pretrained("fuchenru/Trading-Hero-LLM")
nlp = pipeline("text-classification", model=model, tokenizer=tokenizer)
# Preprocess the input text
def preprocess(text, tokenizer, max_length=128):
    inputs = tokenizer(text, truncation=True, padding='max_length', max_length=max_length, return_tensors='pt')
    return inputs

# Function to perform prediction
def predict_sentiment(input_text):
    # Tokenize the input text
    inputs = tokenizer(input_text, return_tensors="pt", truncation=True, padding=True)

    # Perform inference
    with torch.no_grad():
        outputs = model(**inputs)

    # Get predicted label
    predicted_label = torch.argmax(outputs.logits, dim=1).item()

    # Map the predicted label to the original labels
    label_map = {0: 'neutral', 1: 'positive', 2: 'negative'}
    predicted_sentiment = label_map[predicted_label]

    return predicted_sentiment

stock_news = [
    "Market analysts predict a stable outlook for the coming weeks.",
    "The market remained relatively flat today, with minimal movement in stock prices.",
    "Investor sentiment improved following news of a potential trade deal.",
    # .......
]


for i in stock_news:
    predicted_sentiment = predict_sentiment(i)
    print("Predicted Sentiment:", predicted_sentiment)

Advanced Usage

No advanced usage code is provided in the original document, so this part is skipped.

📚 Documentation

Training Data

Evaluation

• Test Accuracy = 0.908469
• Test Precision = 0.927788
• Test Recall = 0.908469
• Test F1 = 0.913267
• Labels: 0 -> Neutral; 1 -> Positive; 2 -> Negative

📄 License

This model is released under the MIT license.

📚 Citation

@misc{yang2020finbert,
    title={FinBERT: A Pretrained Language Model for Financial Communications},
    author={Yi Yang and Mark Christopher Siy UY and Allen Huang},
    year={2020},
    eprint={2006.08097},
    archivePrefix={arXiv},
    }