gavindiaz/NexQuant
1
0
Fork 0
mirror of https://github.com/NicolasBohn/NexQuant.git synced 2026-08-06 19:47:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: rename project from Predix to NexQuant

Browse Source 
Rename all source files, scripts, tests, documentation, and configuration
from Predix/predix to NexQuant/nexquant across the entire codebase.
This commit is contained in:
TPTBusiness
2026-05-09 17:48:22 +02:00
parent 70029b4577
commit 9d6c6d1d5f
92 changed files with 3690 additions and 1015 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/components/coder/kronos_adapter.py
+5 -5
View File
@@ -1,5 +1,5 @@
"""
Kronos Foundation Model Adapter for Predix.
Kronos Foundation Model Adapter for NexQuant.
Wraps the Kronos-mini OHLCV foundation model (4.1M params, AAAI 2026, MIT)
for use as:
@@ -55,8 +55,8 @@ def _ensure_kronos() -> bool:
return _KRONOS_AVAILABLE
def _ohlcv_from_predix(df: pd.DataFrame) -> pd.DataFrame:
"""Convert Predix HDF5 format ($open/$close/...) to Kronos format (open/close/...)."""
def _ohlcv_from_nexquant(df: pd.DataFrame) -> pd.DataFrame:
"""Convert NexQuant HDF5 format ($open/$close/...) to Kronos format (open/close/...)."""
col_map = {"$open": "open", "$high": "high", "$low": "low", "$close": "close", "$volume": "volume"}
renamed = df.rename(columns=col_map)
cols = [c for c in ["open", "high", "low", "close", "volume"] if c in renamed.columns]
@@ -253,7 +253,7 @@ def build_kronos_factor(
instrument = raw.index.get_level_values("instrument").unique()[0]
df = raw.xs(instrument, level="instrument")
ohlcv = _ohlcv_from_predix(df)
ohlcv = _ohlcv_from_nexquant(df)
adapter = KronosAdapter(device=device, max_context=min(context_bars, 512), model_size=model_size)
adapter.load()
@@ -328,7 +328,7 @@ def evaluate_kronos_model(
raw = pd.read_hdf(hdf5_path, key="data")
instrument = raw.index.get_level_values("instrument").unique()[0]
df = raw.xs(instrument, level="instrument")
ohlcv = _ohlcv_from_predix(df)
ohlcv = _ohlcv_from_nexquant(df)
adapter = KronosAdapter(device=device, max_context=min(context_bars, 512), model_size=model_size)
adapter.load()