Todlong / config.py
cwpkd's picture
Create config.py
dd3fbab verified
Raw
History Blame
772 Bytes
# config.py
"""
Configuration file for Yahoo Finance Sentiment Analysis
"""
# Model configurations
FINBERT_MODEL = "ProsusAI/finbert"
GEMMA_MODEL = "google/gemma-2-2b-it" # ใช้ gemma-2-2b แทน 3-4b เนื่องจาก 3-4b ยังไม่มีใน HF
# Yahoo Finance configurations
YAHOO_FINANCE_BASE_URL = "https://finance.yahoo.com"
USER_AGENT = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36"
# Sentiment thresholds
SENTIMENT_THRESHOLDS = {
"positive": 0.05,
"negative": -0.05
}
# Available stock symbols for quick selection
POPULAR_STOCKS = [
"AAPL", "GOOGL", "MSFT", "AMZN", "TSLA",
"META", "NVDA", "JPM", "V", "WMT"
]
# LLM generation parameters
LLM_MAX_LENGTH = 512
LLM_TEMPERATURE = 0.7
LLM_TOP_P = 0.9