This skill utilizes the Kronos model (via KronosPredictorUtility) to perform time-series forecasting and adjust predictions based on news sentiment.
Workflow:
scripts/kronos_predictor.py (via KronosPredictorUtility) to generate the technical/quantitative forecast.references/PROMPTS.md to subjectively adjust the numbers based on latest news/logic.Key Tools:
KronosPredictorUtility.get_base_forecast(df, lookback, pred_len, news_text): Returns List[KLinePoint].Example Usage (Python):
from scripts.utils.kronos_predictor import KronosPredictorUtility
from scripts.utils.database_manager import DatabaseManager
db = DatabaseManager()
predictor = KronosPredictorUtility()
# Forecast
forecast = predictor.predict("600519", horizon="7d")
print(forecast)
This skill requires the Kronos model and an embedding model.
exports/models directory exists in the project root.kronos_news_v1.pt) in exports/models/.[!CAUTION] Model Security: This skill loads model weights from
exports/models. We useweights_only=Trueand only scan for thekronos_news_*.ptpattern. Ensure you only place trusted checkpoints in this directory.
EMBEDDING_MODEL: Path or name of the embedding model (default: sentence-transformers/all-MiniLM-L6-v2).KRONOS_MODEL_PATH: Optional path to override model loading.torch
transformers
sentence-transformers
pandas
numpy
scikit-learn