File size: 796 Bytes
b0d9ebf
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
"""ml_predictor β€” standalone supervised ML price-prediction model for NSE equities.

A quantile-regression predictor (sklearn HistGradientBoosting) that outputs, per
timeframe (INTRADAY / 1D / 3D): a price estimate, buy-price suggestion, stop-loss,
and trend (BULLISH / BEARISH / NEUTRAL). Independent of the LLM debate pipeline.

Modules:
  features.py  β€” shared point-in-time numeric feature builder (all indicators + ML sub-features)
  dataset.py   β€” build training_data.csv from the ohlcv_cache.db (offline)
  train.py     β€” fit the 15 HistGBM estimators + manifest.json
  infer.py     β€” MLPredictor: predict(ticker, tf) / predict_all_tf(ticker)

See CLAUDE.md and the plan for the full design.
"""

from .features import FEATURE_COLUMNS, TIMEFRAMES, compute_features  # noqa: F401