INNER CODE UNIT · Python

_artifacts_dir

initial-d/ml-quant-trading · src/mlquant/cli/main.py:46

def _artifacts_dir(cfg) -> Path:
    p = Path(cfg.get("artifacts_dir", "artifacts"))
    p.mkdir(parents=True, exist_ok=True)
    return p


_PANEL_FIELDS = ("open", "high", "low", "close", "volume", "vwap", "mask")
_OPTIONAL_PANEL_FIELDS = ("amount", "limit_up", "limit_down", "last_close")


def _load_panel(art: Path):
    """Rebuild the :class:`Panel` written by ``gen-data``, optional fields included."""
    from ..data.panel import Panel

    blob = torch.load(art / "panel.pt", weights_only=False)
    optional = {k: blob[k] for k in _OPTIONAL_PANEL_FIELDS if blob.get(k) is not None}
    return Panel(
        dates=blob["dates"], stocks=blob["stocks"],

View source record →