start experiment 49 (exp/49-q18-optimalstopcontrol-entryexit-thresho)

2026-08-20 04:32:01 +00:00
parent fd5382caa4
commit 7698821bbe
27 changed files with 2220 additions and 0 deletions
@@ -0,0 +1,11 @@
+from . import data  # noqa: F401  (registers tac_qlib.contrib.data)
+from . import model, strategy  # noqa: F401
+from .data import TACHandler  # noqa: F401
+from .model import RankICLGBModel  # noqa: F401
+from .strategy import OptimalStopControl  # noqa: F401
+
+__all__ = [
+    "TACHandler",
+    "RankICLGBModel",
+    "OptimalStopControl",
+]
@@ -0,0 +1,3 @@
+from .handler import TACHandler
+
+__all__ = ["TACHandler"]
@@ -0,0 +1,254 @@
+"""TACHandler: a qlib DataHandlerLP that builds datasets from the TradeAC lake.
+
+This is the "custom DataHandler" entry point (Option B): the handler is referenced from the
+workflow yaml's ``dataset.handler`` and reads OHLCV + pre-computed ta-lib features straight
+from the lake parquet files through ``QLibDataLoader`` + the tac_qlib feature provider.
+
+The standard qlib processor pipeline (``infer_processors`` / ``learn_processors``) still runs
+on top, so existing recipes such as ``DropnaLabel``, ``CSZScoreNorm`` or ``RobustZScoreNorm``
+keep working unchanged.
+"""
+
+from __future__ import annotations
+
+import os
+from inspect import getfullargspec
+from typing import List, Optional, Tuple, Union
+
+from qlib.data.dataset import processor as processor_module
+from qlib.data.dataset.handler import DataHandlerLP
+from qlib.utils import get_callable_kwargs
+
+from ...data.config import (
+    LakeConfig,
+    timeframe_for_freq,
+    NON_FEATURE_COLUMNS,
+)
+
+DEFAULT_INFER_PROCESSORS = [
+    {"class": "DropAllNaN", "kwargs": {}},
+    {"class": "ProcessInf", "kwargs": {}},
+    {"class": "ZScoreNorm", "kwargs": {}},
+    {"class": "Fillna", "kwargs": {}},
+]
+DEFAULT_LEARN_PROCESSORS = [
+    {"class": "DropnaLabel"},
+    {"class": "CSZScoreNorm", "kwargs": {"fields_group": "label"}},
+]
+
+#: always include raw OHLCV; ta-lib columns are discovered from the lake and appended.
+RAW_FEATURE_FIELDS = ("$open", "$high", "$low", "$close", "$vwap", "$volume")
+
+DEFAULT_LABEL = "Ref($close,-2)/Ref($close,-1)-1"
+
+
+def check_transform_proc(proc_l, fit_start_time, fit_end_time):
+    """Port of ``qlib.contrib.data.handler.check_transform_proc`` (inject fit window into procs)."""
+    new_l = []
+    for p in proc_l:
+        if not isinstance(p, processor_module.Processor):
+            klass, pkwargs = get_callable_kwargs(p, processor_module)
+            args = getfullargspec(klass).args
+            if "fit_start_time" in args and "fit_end_time" in args:
+                assert fit_start_time is not None and fit_end_time is not None, (
+                    "Make sure `fit_start_time` and `fit_end_time` are not None."
+                )
+                pkwargs.update({"fit_start_time": fit_start_time, "fit_end_time": fit_end_time})
+            proc_config = {"class": klass.__name__, "kwargs": pkwargs}
+            if isinstance(p, dict) and "module_path" in p:
+                proc_config["module_path"] = p["module_path"]
+            new_l.append(proc_config)
+        else:
+            new_l.append(p)
+    return new_l
+
+
+def get_common_feature_fields(lake_root=None, market="US", timeframe="1d") -> List[str]:
+    """Discover feature columns present in *every* feature file of the lake.
+
+    Walks the `family=ta|sp` partition layout (plus any legacy flat files).
+    TA and SP columns are disjoint by construction, so the common set is
+    computed per family (columns shared by all symbol files of that family),
+    then the per-family results are unioned. Returns sorted field names
+    (without the ``$`` prefix). Empty if no features are persisted.
+    """
+    cfg = LakeConfig(lake_root, market)
+    feat_dir = cfg.features_dir(timeframe)
+    if not feat_dir.exists():
+        return []
+    import pyarrow.parquet as pq
+
+    def _family_common(fam_dir: Path) -> set:
+        common = None
+        for p in sorted(fam_dir.glob("symbol=*.parquet")):
+            try:
+                cols = set(pq.read_schema(p).names) - set(NON_FEATURE_COLUMNS)
+            except Exception:  # pragma: no cover - skip unreadable files
+                continue
+            common = cols if common is None else (common & cols)
+            if not common:
+                break
+        return common or set()
+
+    common: set = set()
+    # family tier: features/market=*/timeframe=*/family=*/symbol=*.parquet
+    for fam in ("ta", "sp"):
+        fam_dir = feat_dir / f"family={fam}"
+        if fam_dir.is_dir():
+            common |= _family_common(fam_dir)
+    # legacy flat: features/market=*/timeframe=*/symbol=*.parquet
+    if (feat_dir / "family=ta").exists() or (feat_dir / "family=sp").exists():
+        pass  # family layout already covered
+    else:
+        common |= _family_common(feat_dir)
+    return sorted(common)
+
+
+class DropAllNaN(processor_module.Processor):
+    """Drop feature columns that are all-NaN over the fit window.
+
+    The lake can hold fully-empty indicator columns (e.g. a ta-lib output that was NaN
+    from the start). Such columns carry no learnable signal and make ``ZScoreNorm.fit``
+    warn on empty slices, so we drop them before any other processor runs. The drop set
+    is fixed on the fit window once (during ``fit``), then applied consistently to every
+    segment so train/valid/test keep identical feature columns.
+    """
+
+    def __init__(self, fit_start_time=None, fit_end_time=None):
+        self.fit_start_time = fit_start_time
+        self.fit_end_time = fit_end_time
+        self.cols_to_drop = []
+
+    def fit(self, df=None):
+        if df is None or len(df) == 0:
+            return self
+        window = df
+        if self.fit_start_time is not None and self.fit_end_time is not None:
+            try:
+                from qlib.data.dataset.utils import fetch_df_by_index
+
+                window = fetch_df_by_index(
+                    df, slice(self.fit_start_time, self.fit_end_time), level="datetime"
+                )
+            except Exception:  # pragma: no cover - defensive
+                window = df
+        if len(window) == 0:
+            return self
+        self.cols_to_drop = [c for c in window.columns if window[c].isna().all()]
+        return self
+
+    def __call__(self, df):
+        if self.cols_to_drop:
+            return df.drop(columns=self.cols_to_drop, errors="ignore")
+        return df
+
+
+class TACHandler(DataHandlerLP):
+    """DataHandlerLP backed by the TradeAC parquet lake.
+
+    Parameters mirror ``Alpha158``: ``instruments``/``start_time``/``end_time``/``freq`` define
+    the queried window; ``feature_fields`` selects the features (default: raw OHLCV + all common
+    ta-lib columns found in the lake); ``label`` is a qlib expression for the target.
+    """
+
+    def __init__(
+        self,
+        instruments="all",
+        start_time=None,
+        end_time=None,
+        freq="day",
+        infer_processors=DEFAULT_INFER_PROCESSORS,
+        learn_processors=DEFAULT_LEARN_PROCESSORS,
+        fit_start_time=None,
+        fit_end_time=None,
+        process_type=DataHandlerLP.PTYPE_A,
+        filter_pipe=None,
+        feature_fields=None,
+        label=DEFAULT_LABEL,
+        lake_root=None,
+        market="US",
+        **kwargs,
+    ):
+        # default the processor fit window to the queried window (like Alpha158 without a split)
+        if fit_start_time is None:
+            fit_start_time = start_time
+        if fit_end_time is None:
+            fit_end_time = end_time
+
+        infer_processors = check_transform_proc(infer_processors, fit_start_time, fit_end_time)
+        learn_processors = check_transform_proc(learn_processors, fit_start_time, fit_end_time)
+
+        feature_fields = self._normalize_feature_fields(feature_fields, freq, lake_root, market)
+        if not feature_fields:
+            raise ValueError(
+                "no feature fields available for the lake; set `feature_fields` explicitly "
+                "(e.g. ['$close', '$rsi_14', '$sma_20'])"
+            )
+
+        label_expr, label_names = self._normalize_label(label)
+
+        data_loader = {
+            "class": "QlibDataLoader",
+            "kwargs": {
+                "config": {
+                    "feature": (feature_fields, feature_fields),
+                    "label": (label_expr, label_names),
+                },
+                "filter_pipe": filter_pipe,
+                "freq": freq,
+            },
+        }
+        super().__init__(
+            instruments=instruments,
+            start_time=start_time,
+            end_time=end_time,
+            data_loader=data_loader,
+            infer_processors=infer_processors,
+            learn_processors=learn_processors,
+            process_type=process_type,
+            **kwargs,
+        )
+
+    # ------------------------------------------------------------------ config
+    @staticmethod
+    def _normalize_feature_fields(feature_fields, freq, lake_root, market) -> List[str]:
+        if feature_fields is None:
+            common = get_common_feature_fields(lake_root, market, timeframe_for_freq(freq))
+            feature_fields = list(RAW_FEATURE_FIELDS) + ["$" + f for f in common if "$" + f not in RAW_FEATURE_FIELDS]
+        elif isinstance(feature_fields, str):
+            feature_fields = [f.strip() for f in feature_fields.split(",") if f.strip()]
+        fields = [f if f.startswith("$") else "$" + f for f in feature_fields]
+        # de-dup while preserving order
+        seen, out = set(), []
+        for f in fields:
+            if f not in seen:
+                seen.add(f)
+                out.append(f)
+        return out
+
+    @staticmethod
+    def _normalize_label(label) -> Tuple[List[str], List[str]]:
+        if isinstance(label, str):
+            return [label], ["LABEL0"]
+        if isinstance(label, (list, tuple)):
+            if len(label) == 2 and isinstance(label[0], str):
+                return [label[0]], list(label[1]) if isinstance(label[1], (list, tuple)) else [label[1]]
+            return list(label), ["LABEL%d" % i for i in range(len(label))]
+        raise TypeError(f"unsupported label config: {label!r}")
+
+    # ------------------------------------------------------------------ utils
+    def get_label_config(self):
+        return DEFAULT_LABEL
+
+    @staticmethod
+    def discover_feature_fields(lake_root=None, market="US", freq="day") -> List[str]:
+        return get_common_feature_fields(lake_root, market, timeframe_for_freq(freq))
+
+
+__all__ = ["TACHandler", "DropAllNaN", "get_common_feature_fields"]
+
+
+# Make `DropAllNaN` resolvable by bare name from processor configs (e.g. the default
+# ``infer_processors`` and workflow yamls that reference it without a ``module_path``),
+# mirroring how qlib registers its own processors in ``qlib.data.dataset.processor``.
+processor_module.DropAllNaN = DropAllNaN
@@ -0,0 +1,4 @@
+from .rank_ensemble import RankICEnsembleLGBModel  # noqa: F401
+from .rank_gbdt import RankICLGBModel, rankic_feval  # noqa: F401
+
+__all__ = ["RankICLGBModel", "rankic_feval", "RankICEnsembleLGBModel"]
@@ -0,0 +1,189 @@
+"""Seed-ensembled LightGBM that early-stops on cross-sectional RankIC.
+
+``RankICEnsembleLGBModel`` wraps ``RankICLGBModel`` (per-day RankIC feval +
+``metric='None'`` + ``first_metric_only`` early stopping) over a seed ensemble:
+one sub-model is trained per seed with identical hyper-parameters, and
+predictions are averaged across seeds. This is the model class the
+``tac-rd-rank-ensemble-isolated`` reference run wires into its workflow
+(``module_path: tac_qlib.contrib.model.rank_ensemble``).
+
+The ensemble inherits the RankIC early-stopping behaviour of the single-seed
+model (valid RankIC drives the stopping iteration) while the seed averaging
+stabilizes the prediction against any single seed's early-stopping path.
+
+Training is parallelized: the seed sub-models train in a thread pool —
+``lgb.train`` is C++ and releases the GIL, so concurrent seeds do not block on
+the GIL (5 seeds ~40min/5 on this box). Measured on a 6-physical-core / 12 SMT
+host: the seeds scale ~2x, not linearly — the runs are memory-bandwidth bound
+and each Booster caps its threads at ``cores // workers`` so 5 concurrent
+boosters don't oversubscribe; larger-core hosts scale better. The qlib data
+pipeline is warmed once on the calling thread (fills the handler cache), and
+each worker then prepares its **own** ``lgb.Dataset`` (independent handle, so
+no concurrent ``construct()`` on a shared handle — LightGBM's ``Dataset`` is
+not thread-safe to build). qlib's ``R`` recorder is also not thread-safe, so
+the per-seed evaluation curves are logged on the calling thread after the pool
+finishes.
+
+Wired into a workflow yaml like:
+
+    model:
+        class: RankICEnsembleLGBModel
+        module_path: tac_qlib.contrib.model.rank_ensemble
+        kwargs:
+            loss: mse
+            learning_rate: 0.02
+            num_leaves: 31
+            n_estimators: 3000
+            num_boost_round: 3000
+            early_stopping_rounds: 200
+            min_data_in_leaf: 20
+            lambda_l2: 0.5
+            colsample_bytree: 0.8
+            subsample: 0.8
+            subsample_freq: 1
+            reg_alpha: 0.1
+            reg_lambda: 1.0
+            seeds: "42,7,2026,99,123"
+            parallel: 5
+
+Any ``**kwargs`` other than ``seeds``/``parallel`` are forwarded unchanged to
+every ``RankICLGBModel`` sub-model (same params, different ``seed``).
+"""
+
+from __future__ import annotations
+
+import os
+from concurrent.futures import ThreadPoolExecutor
+from typing import List, Optional
+
+import pandas as pd
+
+from qlib.data.dataset import DatasetH
+from qlib.data.dataset.handler import DataHandlerLP
+
+from tac_qlib.contrib.model.rank_gbdt import RankICLGBModel
+
+__all__ = ["RankICEnsembleLGBModel"]
+
+
+class RankICEnsembleLGBModel(RankICLGBModel):
+    """Seed ensemble of RankIC-early-stopping LightGBM models.
+
+    Parameters
+    ----------
+    seeds : comma-separated integers, one sub-model per seed.
+    parallel : number of seeds to train concurrently. ``0`` (default) = auto
+        (all seeds, bounded by the available cores); ``1`` = sequential.
+    **kwargs : forwarded to every ``RankICLGBModel`` sub-model (model
+        hyper-parameters). ``seeds``/``parallel`` are consumed here and not
+        forwarded.
+    """
+
+    def __init__(self, seeds: str = "42", parallel: int = 0, **kwargs):
+        self.seeds = [int(s.strip()) for s in str(seeds).split(",") if s.strip()]
+        if not self.seeds:
+            raise ValueError("seeds must contain at least one integer")
+        self.parallel = int(parallel)
+        # drop seed/parallel handling from the base kwargs, keep everything else
+        self._model_kwargs = dict(kwargs)
+        super().__init__(**self._model_kwargs)
+        self._models: List[RankICLGBModel] = []
+
+    # --------------------------------------------------------------- helpers
+    @staticmethod
+    def _cores() -> int:
+        try:
+            return max(1, len(os.sched_getaffinity(0)))
+        except AttributeError:
+            return max(1, os.cpu_count() or 1)
+
+    def _worker_count(self) -> int:
+        if self.parallel > 0:
+            return min(len(self.seeds), self.parallel)
+        return min(len(self.seeds), self._cores())
+
+    # ------------------------------------------------------------------ fit
+    def fit(
+        self,
+        dataset: DatasetH,
+        num_boost_round: Optional[int] = None,
+        early_stopping_rounds: Optional[int] = None,
+        verbose_eval: int = 20,
+        evals_result=None,
+        reweighter=None,
+        **kwargs,
+    ):
+        """Train one RankICLGBModel per seed and keep them for prediction.
+
+        The qlib data pipeline is warmed once on this thread (handler cache),
+        then each seed sub-model trains in a parallel worker thread on its own
+        ``lgb.Dataset`` (LightGBM releases the GIL in ``lgb.train``). Evals
+        are logged on this thread after the pool (qlib's ``R`` is not
+        thread-safe).
+        """
+        n_round = num_boost_round or self.num_boost_round
+        n_es = early_stopping_rounds or self.early_stopping_rounds
+
+        if len(self.seeds) == 1:
+            m = RankICLGBModel(seed=self.seeds[0], **self._model_kwargs)
+            m.fit(
+                dataset,
+                num_boost_round=n_round,
+                early_stopping_rounds=n_es,
+                verbose_eval=verbose_eval,
+                evals_result=evals_result,
+                reweighter=reweighter,
+                **kwargs,
+            )
+            self._models = [m]
+            return
+
+        # Warm the qlib handler cache once on this thread so the workers'
+        # concurrent prepare() calls only hit cached frames (no first-write race).
+        proto = RankICLGBModel(seed=self.seeds[0], **self._model_kwargs)
+        proto._prepare_data(dataset, reweighter)
+
+        workers = self._worker_count()
+        # Cap per-Booster threads so concurrent seeds don't oversubscribe
+        # (LightGBM's num_threads=0 uses ALL cores per Booster).
+        per_booster = max(1, self._cores() // workers)
+
+        def fit_seed(seed):
+            m = RankICLGBModel(seed=seed, **self._model_kwargs)
+            if workers > 1 and "num_threads" not in m.params:
+                m.params["num_threads"] = per_booster
+            ds_l = m._prepare_data(dataset, reweighter)
+            booster, evals, names = m._train_from_datasets(
+                ds_l,
+                num_boost_round=n_round,
+                early_stopping_rounds=n_es,
+                verbose_eval=verbose_eval,
+                **kwargs,
+            )
+            m.model = booster
+            return m, evals, names
+
+        with ThreadPoolExecutor(max_workers=workers) as ex:
+            results = list(ex.map(fit_seed, self.seeds))
+
+        self._models = [m for m, _, _ in results]
+
+        # Merge + log evals on the main thread (qlib's R is not thread-safe).
+        if evals_result is not None:
+            for m, evals, names in results:
+                for k in names:
+                    for key, val in evals.get(k, {}).items():
+                        evals_result.setdefault(f"{k}.seed{m.params['seed']}", {})[key] = val
+        for m, evals, names in results:
+            self._log_evals(evals, names, prefix=f"seed{m.params['seed']}.")
+
+    # -------------------------------------------------------------- predict
+    def predict(self, dataset: DatasetH, segment="test") -> pd.Series:
+        """Average the per-seed predictions over the given segment."""
+        if not self._models:
+            raise ValueError("model is not fitted yet!")
+        preds = [m.predict(dataset, segment=segment) for m in self._models]
+        if len(preds) == 1:
+            return preds[0]
+        frame = pd.concat(preds, axis=1)
+        return frame.mean(axis=1)
@@ -0,0 +1,238 @@
+"""LGBModel variant that early-stops on cross-sectional RankIC instead of l2.
+
+Standard qlib ``LGBModel`` early-stops on the regression loss (mse). For
+cross-sectional alpha signals the quantity we actually care about is the per-day
+rank correlation (Rank IC), which mse early-stopping does not optimize for.
+Experiments on the 50-ETF lake (SP-5d 55-feature panel) show that early-stopping
+on a custom RankIC feval lifts RankIC 0.047 -> 0.075 vs. the mse-stopped model.
+
+This class reuses ``LGBModel``'s data preparation but:
+
+  - tags each ``lgb.Dataset`` with per-day query ``group`` sizes so a ranking
+    metric can be computed per trading day;
+  - injects a custom ``feval`` (mean per-day Spearman of pred vs label) into
+    ``lgb.train``; early stopping then selects the iteration that maximizes
+    RankIC on the valid set;
+  - forces ``metric='None'`` + ``first_metric_only=True`` so early-stopping
+    tracks RankIC only (not the regression loss).
+
+Wired into a workflow yaml like:
+
+    model:
+        class: RankICLGBModel
+        module_path: tac_qlib.contrib.model.rank_gbdt
+        kwargs:
+            loss: mse
+            learning_rate: 0.03
+            num_leaves: 31
+            n_estimators: 500
+            ...
+
+The rank feval is used for early-stopping selection only; the objective stays
+the configured loss (default mse). Set ``rank_eval=False`` to fall back to the
+plain LGBModel behaviour (early-stop on the loss).
+
+Generic: works for any cross-sectional panel whose qlib dataset index has a
+``datetime`` level (each level value = one query group). The per-day groups are
+derived automatically, so no universe-specific configuration is needed.
+"""
+
+from __future__ import annotations
+
+from typing import List, Optional, Tuple
+
+import numpy as np
+import pandas as pd
+import lightgbm as lgb
+
+from qlib.data.dataset import DatasetH
+from qlib.data.dataset.handler import DataHandlerLP
+from qlib.contrib.model.gbdt import LGBModel
+from qlib.workflow import R
+
+__all__ = ["RankICLGBModel", "rankic_feval"]
+
+
+def _group_averaged_rank(values: np.ndarray, gid: np.ndarray, offs: np.ndarray) -> np.ndarray:
+    """Averaged (tie-corrected) rank of ``values`` within each group, vectorized.
+
+    ``gid`` maps each row to its group id; ``offs`` holds the cumulative row
+    offsets so that group ``i`` occupies rows ``[offs[i], offs[i+1])``. Returns
+    the same result as ``pandas.Series.rank(method='average')`` applied per
+    group, but in one pass (``np.lexsort`` is the only non-linear step).
+    """
+    n = len(values)
+    order = np.lexsort((values, gid))
+    ord_rank = np.empty(n, dtype=np.float64)
+    ord_rank[order] = np.arange(n, dtype=np.float64) - offs[gid[order]] + 1.0
+    sg = gid[order]
+    sv = values[order]
+    newblock = np.empty(n, dtype=bool)
+    newblock[0] = True
+    newblock[1:] = (sg[1:] != sg[:-1]) | (sv[1:] != sv[:-1])
+    blockid = np.cumsum(newblock) - 1
+    block_mean = np.bincount(blockid, weights=ord_rank[order]) / np.bincount(blockid)
+    out = np.empty(n)
+    out[order] = block_mean[blockid]
+    return out
+
+
+def _per_day_spearman(preds: np.ndarray, labels: np.ndarray, group: np.ndarray) -> float:
+    """Mean per-day Spearman rank correlation of preds vs labels.
+
+    ``group`` holds the number of rows of each trading day (query group), in
+    order. Days with <3 valid rows or a constant pred/label are skipped.
+
+    Vectorized: per-day Spearman == Pearson of the per-day rank transforms,
+    and the Pearson moments (``sum``, ``sum`` of products/squares) aggregate
+    over each day with ``np.bincount``. Runs ~10x faster than the per-day
+    ``pd.Series.rank()`` loop that preceded it — this feval is invoked on the
+    train and valid panels every boosting round, per seed.
+    """
+    if group is None or len(group) == 0:
+        return 0.0
+    offs = np.concatenate([[0], np.cumsum(group.astype(int))])
+    gid = np.repeat(np.arange(len(group)), group.astype(int))
+    rp = _group_averaged_rank(preds, gid, offs)
+    rl = _group_averaged_rank(labels, gid, offs)
+    n_g = group.astype(float)
+    s_p = np.bincount(gid, weights=rp)
+    s_l = np.bincount(gid, weights=rl)
+    s_pl = np.bincount(gid, weights=rp * rl)
+    s_pp = np.bincount(gid, weights=rp * rp)
+    s_ll = np.bincount(gid, weights=rl * rl)
+    cov = n_g * s_pl - s_p * s_l
+    var_p = n_g * s_pp - s_p ** 2
+    var_l = n_g * s_ll - s_l ** 2
+    denom = np.sqrt(var_p * var_l)
+    valid = (n_g >= 3) & (denom > 0)
+    corr = np.where(valid, cov / np.where(denom == 0, 1, denom), 0.0)
+    return float(corr[valid].mean()) if valid.any() else 0.0
+
+
+def rankic_feval(preds, dataset):
+    """LightGBM feval: mean RankIC (higher is better in lgb convention)."""
+    labels = dataset.get_label()
+    group = dataset.get_group()
+    ric = _per_day_spearman(preds, labels, group)
+    return "rankic", ric, True  # (name, value, higher_is_better)
+
+
+class RankICLGBModel(LGBModel):
+    """LGBModel that early-stops on per-day RankIC via a custom feval."""
+
+    def __init__(self, rank_eval: bool = True, **kwargs):
+        super().__init__(**kwargs)
+        self.rank_eval = rank_eval
+
+    def _prepare_data(self, dataset: DatasetH, reweighter=None) -> List[Tuple[lgb.Dataset, str]]:
+        ds_l = []
+        assert "train" in dataset.segments
+        for key in ["train", "valid"]:
+            if key in dataset.segments:
+                df = dataset.prepare(key, col_set=["feature", "label"], data_key=DataHandlerLP.DK_L)
+                if df.empty:
+                    raise ValueError("Empty data from dataset, please check your dataset config.")
+                x, y = df["feature"], df["label"]
+                if y.values.ndim == 2 and y.values.shape[1] == 1:
+                    y = np.squeeze(y.values)
+                else:
+                    raise ValueError("LightGBM doesn't support multi-label training")
+
+                if reweighter is None:
+                    w = None
+                elif hasattr(reweighter, "reweight"):
+                    w = reweighter.reweight(df)
+                else:
+                    raise ValueError("Unsupported reweighter type.")
+
+                # per-day query groups: each trading day is one group
+                if self.rank_eval and isinstance(df.index, pd.MultiIndex) and "datetime" in df.index.names:
+                    group = df.groupby(level="datetime").size().to_numpy(dtype=np.int32)
+                else:
+                    group = None
+
+                d = lgb.Dataset(x.values, label=y, weight=w, group=group, free_raw_data=False)
+                ds_l.append((d, key))
+        return ds_l
+
+    def _train_from_datasets(
+        self,
+        ds_l: List[Tuple[lgb.Dataset, str]],
+        num_boost_round: Optional[int] = None,
+        early_stopping_rounds: Optional[int] = None,
+        verbose_eval: int = 20,
+        evals_result=None,
+        **kwargs,
+    ) -> Tuple[lgb.Booster, dict, List[str]]:
+        """Train a Booster from already-prepared ``lgb.Dataset`` objects.
+
+        Pure training — no ``R.log_metrics`` — so it can be called from worker
+        threads (qlib's ``R`` recorder is not thread-safe; the caller decides
+        when/where to log). Returns ``(booster, evals_result, segment_names)``.
+        """
+        if evals_result is None:
+            evals_result = {}
+        ds, names = list(zip(*ds_l))
+
+        callbacks = [
+            lgb.early_stopping(
+                self.early_stopping_rounds if early_stopping_rounds is None else early_stopping_rounds
+            ),
+            lgb.log_evaluation(period=verbose_eval),
+            lgb.record_evaluation(evals_result),
+        ]
+        if self.rank_eval:
+            # early-stopping must be driven ONLY by the RankIC feval, not l2.
+            # metric='None' suppresses the default l2 metric; first_metric_only
+            # makes early_stopping track the single remaining (rankic) metric.
+            self.params["metric"] = "None"
+            self.params["first_metric_only"] = True
+            feval = rankic_feval
+        else:
+            self.params.pop("metric", None)
+            self.params.pop("first_metric_only", None)
+            feval = None
+
+        booster = lgb.train(
+            self.params,
+            ds[0],
+            num_boost_round=self.num_boost_round if num_boost_round is None else num_boost_round,
+            valid_sets=ds,
+            valid_names=names,
+            feval=feval,
+            callbacks=callbacks,
+            **kwargs,
+        )
+        return booster, evals_result, list(names)
+
+    def _log_evals(self, evals_result, names: List[str], prefix: str = "") -> None:
+        """Log recorded evaluation curves to qlib's active recorder."""
+        for k in names:
+            for key, val in evals_result.get(k, {}).items():
+                name = f"{prefix}{key}.{k}"
+                for epoch, m in enumerate(val):
+                    R.log_metrics(**{name.replace("@", "_"): m}, step=epoch)
+
+    def fit(
+        self,
+        dataset: DatasetH,
+        num_boost_round: Optional[int] = None,
+        early_stopping_rounds: Optional[int] = None,
+        verbose_eval: int = 20,
+        evals_result=None,
+        reweighter=None,
+        **kwargs,
+    ):
+        if evals_result is None:
+            evals_result = {}
+        ds_l = self._prepare_data(dataset, reweighter)
+        self.model, evals_result, names = self._train_from_datasets(
+            ds_l,
+            num_boost_round=num_boost_round,
+            early_stopping_rounds=early_stopping_rounds,
+            verbose_eval=verbose_eval,
+            evals_result=evals_result,
+            **kwargs,
+        )
+        self._log_evals(evals_result, names)
@@ -0,0 +1,13 @@
+from .kelly_dropout import FractionalKellyDropoutStrategy  # noqa: F401
+from .optimal_stop import OptimalStopControl  # noqa: F401
+from .regime_gate import RegimeGateDropoutStrategy  # noqa: F401
+from .top_bottom import TopBottomDropoutStrategy  # noqa: F401
+from .weekly_rebalance import WeeklyRebalanceDropoutStrategy  # noqa: F401
+
+__all__ = [
+    "OptimalStopControl",
+    "FractionalKellyDropoutStrategy",
+    "WeeklyRebalanceDropoutStrategy",
+    "TopBottomDropoutStrategy",
+    "RegimeGateDropoutStrategy",
+]
@@ -0,0 +1,201 @@
+"""Fractional-Kelly dropout strategy for cross-sectional signals.
+
+Sizing rule variant of ``qlib.contrib.strategy.signal_strategy.TopkDropoutStrategy``:
+the topk/n_drop SELECTION is identical to the reference, but the buy size is
+proportional to the score MAGNITUDE (edge) instead of equal-weight, capped at a
+fraction ``cap_frac`` of the equal-weight notional so a single name cannot
+over-concentrate the book.
+
+``cap_frac`` is the fraction of the equal-weight per-name notional that a top
+signal can deploy at most (e.g. 0.5 = at most half the equal-weight size).
+Names whose score is below the median of the buy set get a proportionally
+smaller slice; the residual stays in cash (that is the point of the rule:
+throw away less edge per name, deploy less capital when conviction is low).
+"""
+
+from __future__ import annotations
+
+from typing import List
+
+import numpy as np
+import pandas as pd
+
+from qlib.backtest import Order
+from qlib.backtest.decision import OrderDir, TradeDecisionWO
+from qlib.contrib.strategy.signal_strategy import TopkDropoutStrategy
+
+__all__ = ["FractionalKellyDropoutStrategy"]
+
+DEFAULT_CAP_FRAC = 0.5
+
+
+class FractionalKellyDropoutStrategy(TopkDropoutStrategy):
+    """TopkDropout selection with score-magnitude (fractional-Kelly) sizing.
+
+    Parameters
+    ----------
+    topk, n_drop, method_sell, method_buy, hold_thresh, only_tradable,
+    forbid_all_trade_at_limit : same as ``TopkDropoutStrategy``.
+    cap_frac : max buy notional as a fraction of the equal-weight notional.
+    """
+
+    def __init__(self, *, topk, n_drop, cap_frac: float = DEFAULT_CAP_FRAC, **kwargs):
+        super().__init__(topk=topk, n_drop=n_drop, **kwargs)
+        self.cap_frac = cap_frac
+
+    def generate_trade_decision(self, execute_result=None):
+        import copy
+
+        trade_step = self.trade_calendar.get_trade_step()
+        trade_start_time, trade_end_time = self.trade_calendar.get_step_time(trade_step)
+        pred_start_time, pred_end_time = self.trade_calendar.get_step_time(trade_step, shift=1)
+        pred_score = self.signal.get_signal(start_time=pred_start_time, end_time=pred_end_time)
+        if isinstance(pred_score, pd.DataFrame):
+            pred_score = pred_score.iloc[:, 0]
+        if pred_score is None:
+            return TradeDecisionWO([], self)
+
+        if self.only_tradable:
+
+            def get_first_n(li, n, reverse=False):
+                cur_n = 0
+                res = []
+                for si in reversed(li) if reverse else li:
+                    if self.trade_exchange.is_stock_tradable(
+                        stock_id=si, start_time=trade_start_time, end_time=trade_end_time
+                    ):
+                        res.append(si)
+                        cur_n += 1
+                        if cur_n >= n:
+                            break
+                return res[::-1] if reverse else res
+
+            def get_last_n(li, n):
+                return get_first_n(li, n, reverse=True)
+
+            def filter_stock(li):
+                return [
+                    si
+                    for si in li
+                    if self.trade_exchange.is_stock_tradable(
+                        stock_id=si, start_time=trade_start_time, end_time=trade_end_time
+                    )
+                ]
+
+        else:
+
+            def get_first_n(li, n):
+                return list(li)[:n]
+
+            def get_last_n(li, n):
+                return list(li)[-n:]
+
+            def filter_stock(li):
+                return li
+
+        current_temp: "object" = copy.deepcopy(self.trade_position)
+        sell_order_list: List[Order] = []
+        buy_order_list: List[Order] = []
+        cash = current_temp.get_cash()
+        current_stock_list = current_temp.get_stock_list()
+        last = pred_score.reindex(current_stock_list).sort_values(ascending=False).index
+
+        if self.method_buy == "top":
+            today = get_first_n(
+                pred_score[~pred_score.index.isin(last)].sort_values(ascending=False).index,
+                self.n_drop + self.topk - len(last),
+            )
+        elif self.method_buy == "random":
+            topk_candi = get_first_n(pred_score.sort_values(ascending=False).index, self.topk)
+            candi = list(filter(lambda x: x not in last, topk_candi))
+            n = self.n_drop + self.topk - len(last)
+            try:
+                today = np.random.choice(candi, n, replace=False)
+            except ValueError:
+                today = candi
+        else:
+            raise NotImplementedError(f"This type of input is not supported")
+
+        comb = pred_score.reindex(last.union(pd.Index(today))).sort_values(ascending=False).index
+
+        if self.method_sell == "bottom":
+            sell = last[last.isin(get_last_n(comb, self.n_drop))]
+        elif self.method_sell == "random":
+            candi = filter_stock(last)
+            try:
+                sell = pd.Index(np.random.choice(candi, self.n_drop, replace=False) if len(last) else [])
+            except ValueError:
+                sell = candi
+        else:
+            raise NotImplementedError(f"This type of input is not supported")
+
+        buy = today[: len(sell) + self.topk - len(last)]
+        for code in current_stock_list:
+            if not self.trade_exchange.is_stock_tradable(
+                stock_id=code,
+                start_time=trade_start_time,
+                end_time=trade_end_time,
+                direction=None if self.forbid_all_trade_at_limit else OrderDir.SELL,
+            ):
+                continue
+            if code in sell:
+                time_per_step = self.trade_calendar.get_freq()
+                if current_temp.get_stock_count(code, bar=time_per_step) < self.hold_thresh:
+                    continue
+                sell_amount = current_temp.get_stock_amount(code=code)
+                sell_order = Order(
+                    stock_id=code,
+                    amount=sell_amount,
+                    start_time=trade_start_time,
+                    end_time=trade_end_time,
+                    direction=Order.SELL,
+                )
+                if self.trade_exchange.check_order(sell_order):
+                    sell_order_list.append(sell_order)
+                    trade_val, trade_cost, trade_price = self.trade_exchange.deal_order(
+                        sell_order, position=current_temp
+                    )
+                    cash += trade_val - trade_cost
+
+        if len(buy) == 0:
+            return TradeDecisionWO(sell_order_list, self)
+
+        # ---- fractional-Kelly sizing --------------------------------------
+        # equal-weight notional (reference baseline)
+        eq_notional = cash * self.risk_degree / len(buy)
+        buy_scores = pred_score.reindex(buy).astype(float)
+        lo, hi = buy_scores.min(), buy_scores.max()
+        if hi == lo:
+            w = pd.Series(1.0, index=buy_scores.index)
+        else:
+            w = (buy_scores - lo) / (hi - lo)  # [0,1] edge magnitude
+            w = w.clip(lower=0.0)
+            w_max = w.max()
+            w = w / w_max if w_max > 0 else w  # max == 1.0
+        for code in buy:
+            if not self.trade_exchange.is_stock_tradable(
+                stock_id=code,
+                start_time=trade_start_time,
+                end_time=trade_end_time,
+                direction=None if self.forbid_all_trade_at_limit else OrderDir.BUY,
+            ):
+                continue
+            buy_price = self.trade_exchange.get_deal_price(
+                stock_id=code, start_time=trade_start_time, end_time=trade_end_time, direction=OrderDir.BUY
+            )
+            notional = eq_notional * min(self.cap_frac, float(w.get(code, 0.0)))
+            buy_amount = notional / buy_price
+            factor = self.trade_exchange.get_factor(
+                stock_id=code, start_time=trade_start_time, end_time=trade_end_time
+            )
+            buy_amount = self.trade_exchange.round_amount_by_trade_unit(buy_amount, factor)
+            buy_order = Order(
+                stock_id=code,
+                amount=buy_amount,
+                start_time=trade_start_time,
+                end_time=trade_end_time,
+                direction=Order.BUY,
+            )
+            buy_order_list.append(buy_order)
+
+        return TradeDecisionWO(sell_order_list + buy_order_list, self)
@@ -0,0 +1,217 @@
+"""Optimal-stopping / stochastic-control strategy for cross-sectional signals.
+
+Entry is a control policy: a symbol opens a position only when its cross-sectional
+signal percentile is at or above ``entry_pct`` (i.e. it is one of the top-ranked
+names) and the portfolio has fewer than ``topk`` open positions.
+
+Exit is an optimal-stopping rule: a held position is stopped (closed) when its
+signal percentile falls below ``exit_pct`` (the continuation value of holding is
+no longer worth the risk), OR after ``max_hold_days`` (time stop / finite
+horizon), OR when the position P&L breaches ``sl`` (loss control) and the
+position has been held at least ``min_hold_days``.
+
+Sizing is fixed ``notional`` per position (equal-weight control), unlike the
+TopkDropout cash-allocation heuristic.
+
+Wired into qrun workflows like any ``BaseStrategy`` (see ``PortAnaRecord``
+config). Mirrors the API usage of qlib's ``TopkDropoutStrategy``: ``Order``/
+``OrderDir`` from ``qlib.backtest.decision``, ``trade_calendar`` /
+``trade_exchange`` / ``trade_position`` injected by the backtest executor.
+"""
+
+from __future__ import annotations
+
+from typing import List
+
+import pandas as pd
+
+from qlib.backtest import Order
+from qlib.backtest.decision import OrderDir, TradeDecisionWO
+from qlib.contrib.strategy.signal_strategy import BaseSignalStrategy
+
+__all__ = ["OptimalStopControl"]
+
+DEFAULT_NOTIONAL = 20_000.0
+DEFAULT_ENTRY_PCT = 0.80
+DEFAULT_EXIT_PCT = 0.50
+DEFAULT_MAX_HOLD_DAYS = 10
+DEFAULT_MIN_HOLD_DAYS = 2
+DEFAULT_SL = -0.06
+
+
+class OptimalStopControl(BaseSignalStrategy):
+    """Optimal-stopping long-only strategy over a cross-sectional signal.
+
+    Parameters
+    ----------
+    topk : max number of concurrent positions.
+    entry_pct : min cross-sectional score percentile required to OPEN (0..1).
+    exit_pct : held positions are stopped when score percentile < exit_pct.
+    max_hold_days : hard time stop (finite-horizon close).
+    min_hold_days : minimum holding days before stop-loss is evaluated.
+    notional : $ per position (equal-weight control).
+    sl : stop-loss threshold as fraction of entry price (<= 0), disabled if 0.
+    """
+
+    def __init__(
+        self,
+        *,
+        signal=None,
+        topk: int = 10,
+        entry_pct: float = DEFAULT_ENTRY_PCT,
+        exit_pct: float = DEFAULT_EXIT_PCT,
+        max_hold_days: int = DEFAULT_MAX_HOLD_DAYS,
+        min_hold_days: int = DEFAULT_MIN_HOLD_DAYS,
+        notional: float = DEFAULT_NOTIONAL,
+        sl: float = DEFAULT_SL,
+        risk_degree: float = 0.95,
+        trade_exchange=None,
+        level_infra=None,
+        common_infra=None,
+        **kwargs,
+    ):
+        super().__init__(
+            signal=signal,
+            trade_exchange=trade_exchange,
+            level_infra=level_infra,
+            common_infra=common_infra,
+            **kwargs,
+        )
+        self.topk = topk
+        self.entry_pct = entry_pct
+        self.exit_pct = exit_pct
+        self.max_hold_days = max_hold_days
+        self.min_hold_days = min_hold_days
+        self.notional = notional
+        self.sl = sl
+
+    # ------------------------------------------------------------------ utils
+    @staticmethod
+    def _pct_rank(score: pd.Series) -> pd.Series:
+        return score.rank(pct=True)
+
+    def _entry_price(self, pos) -> float:
+        # Position stores avg entry price under key "price" (see Position.position)
+        price = pos.position.get("price")
+        if price is None:
+            price = pos.get_stock_amount("price")
+        return float(price)
+
+    def _pnl_pct(self, pos, mark: float) -> float:
+        entry = self._entry_price(pos)
+        if not entry or entry != entry:
+            return 0.0
+        return mark / entry - 1.0
+
+    def _is_tradable(self, code, start, end, direction) -> bool:
+        try:
+            return self.trade_exchange.is_stock_tradable(
+                stock_id=code, start_time=start, end_time=end, direction=direction
+            )
+        except TypeError:  # some exchanges take no direction kwarg
+            return self.trade_exchange.is_stock_tradable(stock_id=code, start_time=start, end_time=end)
+
+    # ------------------------------------------------------------ decision
+    def generate_trade_decision(self, execute_result=None):
+        trade_step = self.trade_calendar.get_trade_step()
+        trade_start, trade_end = self.trade_calendar.get_step_time(trade_step)
+        pred_start, pred_end = self.trade_calendar.get_step_time(trade_step, shift=1)
+        pred_score = self.signal.get_signal(start_time=pred_start, end_time=pred_end)
+        if isinstance(pred_score, pd.DataFrame):
+            pred_score = pred_score.iloc[:, 0]
+        if pred_score is None or len(pred_score) == 0:
+            return TradeDecisionWO([], self)
+
+        pct = self._pct_rank(pred_score)
+        time_per_step = self.trade_calendar.get_freq()
+        current_temp = __import__("copy").deepcopy(self.trade_position)
+
+        holdings = {}
+        for code in current_temp.get_stock_list():
+            if abs(current_temp.get_stock_amount(code)) > 1e-6:
+                holdings[code] = current_temp
+
+        # ---- optimal stopping: close held positions -----------------------
+        sell_orders: List[Order] = []
+        closed_today = set()
+        kept = {}
+        for code, pos in holdings.items():
+            held = current_temp.get_stock_count(code, bar=time_per_step)
+            mark = self.trade_exchange.get_deal_price(
+                stock_id=code, start_time=trade_start, end_time=trade_end, direction=Order.SELL
+            )
+            if mark is None or mark != mark:
+                continue
+            rank = pct.get(code, 0.0)
+            stop_pnl = held >= self.min_hold_days and self.sl < 0 and self._pnl_pct(pos, mark) <= self.sl
+            if held >= self.max_hold_days or rank < self.exit_pct or stop_pnl:
+                amt = abs(current_temp.get_stock_amount(code))
+                o = Order(stock_id=code, amount=amt, start_time=trade_start,
+                          end_time=trade_end, direction=Order.SELL)
+                if self.trade_exchange.check_order(o):
+                    sell_orders.append(o)
+                    self.trade_exchange.deal_order(o, position=current_temp)
+                closed_today.add(code)
+            else:
+                kept[code] = mark
+
+        # ---- equal-weight control: target notional per name -----------------
+        # candidate opens: top-ranked names whose signal pct >= entry_pct
+        rank_desc = pred_score.sort_values(ascending=False)
+        held_codes = set(kept)
+        opens = []
+        for sym in rank_desc.index:
+            if len(opens) >= self.topk:
+                break
+            if sym in held_codes:
+                continue
+            if pct.get(sym, 0.0) < self.entry_pct:
+                continue
+            if not self._is_tradable(sym, trade_start, trade_end, OrderDir.BUY):
+                continue
+            opens.append(sym)
+
+        targets = held_codes | set(opens)
+        if not targets:
+            return TradeDecisionWO(sell_orders, self)
+
+        # total value (cash + marked positions) -> per-target notional
+        total_value = current_temp.get_cash()
+        for code, mark in kept.items():
+            total_value += abs(current_temp.get_stock_amount(code)) * mark
+
+        target_notional = total_value * self.risk_degree / max(1, len(targets))
+
+        # ---- rebalance kept positions toward target weight ------------------
+        buy_orders: List[Order] = []
+        for code, mark in kept.items():
+            cur = abs(current_temp.get_stock_amount(code)) * mark
+            diff_notional = target_notional - cur
+            if abs(diff_notional) / target_notional < 0.02:
+                continue  # skip tiny rebalances
+            amount_delta = diff_notional / mark
+            direction = Order.BUY if amount_delta > 0 else Order.SELL
+            o = Order(stock_id=code, amount=abs(amount_delta), start_time=trade_start,
+                      end_time=trade_end, direction=direction)
+            if self.trade_exchange.check_order(o):
+                (buy_orders if direction == Order.BUY else sell_orders).append(o)
+                self.trade_exchange.deal_order(o, position=current_temp)
+
+        # ---- open new positions at target weight ----------------------------
+        for sym in opens:
+            px = self.trade_exchange.get_deal_price(
+                stock_id=sym, start_time=trade_start, end_time=trade_end, direction=OrderDir.BUY
+            )
+            if px is None or px != px or px <= 0:
+                continue
+            amount = target_notional / px
+            factor = self.trade_exchange.get_factor(
+                stock_id=sym, start_time=trade_start, end_time=trade_end
+            )
+            amount = self.trade_exchange.round_amount_by_trade_unit(amount, factor)
+            o = Order(stock_id=sym, amount=amount, start_time=trade_start,
+                      end_time=trade_end, direction=Order.BUY)
+            if self.trade_exchange.check_order(o):
+                buy_orders.append(o)
+
+        return TradeDecisionWO(sell_orders + buy_orders, self)
@@ -0,0 +1,231 @@
+"""HMM-regime overlay TopkDropout strategy.
+
+Regime-gate overlay on ``qlib.contrib.strategy.signal_strategy.TopkDropoutStrategy``:
+selection and sizing are identical to the reference, but a name is only BOUGHT
+(entry gate) when its per-symbol HMM regime posterior ``sp_hmm_p_regime1`` on
+the signal date is >= ``regime_threshold``; otherwise it is held in cash instead
+of being opened.
+
+The regime posterior is read from the lake feature provider on the fly via
+``qlib.data.D.features`` (field ``$sp_hmm_p_regime1``) for the signal window, so
+no regime column needs to enter the model's ``feature_fields`` — the gate is a
+pure overlay (book ch.01: regime flags regressed as model features, survived
+only as an overlay). The HMM itself was fit with ``fit_end=<train end>`` when
+the lake features were backfilled, so there is no lookahead.
+
+Names already held are NOT force-sold when the regime turns unfavourable
+(entry gate only, matching the queue-10 design).
+"""
+
+from __future__ import annotations
+
+from typing import List
+
+import numpy as np
+import pandas as pd
+
+from qlib.backtest import Order
+from qlib.backtest.decision import OrderDir, TradeDecisionWO
+from qlib.contrib.strategy.signal_strategy import TopkDropoutStrategy
+
+try:
+    from qlib.data import D
+except ImportError:  # pragma: no cover - qlib always present in this stack
+    D = None
+
+__all__ = ["RegimeGateDropoutStrategy"]
+
+DEFAULT_REGIME_THRESHOLD = 0.5
+REGIME_FIELD = "$sp_hmm_p_regime1"
+
+
+class RegimeGateDropoutStrategy(TopkDropoutStrategy):
+    """TopkDropout with an HMM-regime entry gate on buy candidates.
+
+    Parameters
+    ----------
+    topk, n_drop, method_sell, method_buy, hold_thresh, only_tradable,
+    forbid_all_trade_at_limit : same as ``TopkDropoutStrategy``.
+    regime_threshold : minimum ``sp_hmm_p_regime1`` posterior required to open a
+        new position (default 0.5).
+    """
+
+    def __init__(self, *, topk, n_drop, regime_threshold: float = DEFAULT_REGIME_THRESHOLD, **kwargs):
+        super().__init__(topk=topk, n_drop=n_drop, **kwargs)
+        self.regime_threshold = regime_threshold
+
+    def _regime_for(self, codes, pred_start, pred_end) -> pd.Series:
+        """Return {code: sp_hmm_p_regime1} for the signal window (last day)."""
+        if D is None:
+            return pd.Series(dtype=float)
+        try:
+            df = D.features(list(codes), [REGIME_FIELD], start_time=pred_start, end_time=pred_end, freq="day")
+        except Exception:  # noqa: BLE001 - a regime read failure should gate open, not crash
+            return pd.Series(dtype=float)
+        if df is None or len(df) == 0:
+            return pd.Series(dtype=float)
+        # df index is MultiIndex (datetime, instrument); take the last day's values
+        df = df.reset_index()
+        ts_col = "datetime" if "datetime" in df.columns else df.columns[0]
+        sym_col = "instrument" if "instrument" in df.columns else df.columns[1]
+        last_ts = df[ts_col].max()
+        last = df[df[ts_col] == last_ts]
+        out = {}
+        for _, row in last.iterrows():
+            sym = str(row[sym_col]).split("/")[-1].upper()
+            val = row.iloc[-1]
+            out[sym] = float(val) if val == val else np.nan
+        return pd.Series(out)
+
+    def generate_trade_decision(self, execute_result=None):
+        import copy
+
+        trade_step = self.trade_calendar.get_trade_step()
+        trade_start_time, trade_end_time = self.trade_calendar.get_step_time(trade_step)
+        pred_start_time, pred_end_time = self.trade_calendar.get_step_time(trade_step, shift=1)
+        pred_score = self.signal.get_signal(start_time=pred_start_time, end_time=pred_end_time)
+        if isinstance(pred_score, pd.DataFrame):
+            pred_score = pred_score.iloc[:, 0]
+        if pred_score is None:
+            return TradeDecisionWO([], self)
+
+        if self.only_tradable:
+
+            def get_first_n(li, n, reverse=False):
+                cur_n = 0
+                res = []
+                for si in reversed(li) if reverse else li:
+                    if self.trade_exchange.is_stock_tradable(
+                        stock_id=si, start_time=trade_start_time, end_time=trade_end_time
+                    ):
+                        res.append(si)
+                        cur_n += 1
+                        if cur_n >= n:
+                            break
+                return res[::-1] if reverse else res
+
+            def get_last_n(li, n):
+                return get_first_n(li, n, reverse=True)
+
+            def filter_stock(li):
+                return [
+                    si
+                    for si in li
+                    if self.trade_exchange.is_stock_tradable(
+                        stock_id=si, start_time=trade_start_time, end_time=trade_end_time
+                    )
+                ]
+
+        else:
+
+            def get_first_n(li, n):
+                return list(li)[:n]
+
+            def get_last_n(li, n):
+                return list(li)[-n:]
+
+            def filter_stock(li):
+                return li
+
+        current_temp: "object" = copy.deepcopy(self.trade_position)
+        sell_order_list: List[Order] = []
+        buy_order_list: List[Order] = []
+        cash = current_temp.get_cash()
+        current_stock_list = current_temp.get_stock_list()
+        last = pred_score.reindex(current_stock_list).sort_values(ascending=False).index
+
+        if self.method_buy == "top":
+            today = get_first_n(
+                pred_score[~pred_score.index.isin(last)].sort_values(ascending=False).index,
+                self.n_drop + self.topk - len(last),
+            )
+        elif self.method_buy == "random":
+            topk_candi = get_first_n(pred_score.sort_values(ascending=False).index, self.topk)
+            candi = list(filter(lambda x: x not in last, topk_candi))
+            n = self.n_drop + self.topk - len(last)
+            try:
+                today = np.random.choice(candi, n, replace=False)
+            except ValueError:
+                today = candi
+        else:
+            raise NotImplementedError(f"This type of input is not supported")
+
+        comb = pred_score.reindex(last.union(pd.Index(today))).sort_values(ascending=False).index
+
+        if self.method_sell == "bottom":
+            sell = last[last.isin(get_last_n(comb, self.n_drop))]
+        elif self.method_sell == "random":
+            candi = filter_stock(last)
+            try:
+                sell = pd.Index(np.random.choice(candi, self.n_drop, replace=False) if len(last) else [])
+            except ValueError:
+                sell = candi
+        else:
+            raise NotImplementedError(f"This type of input is not supported")
+
+        buy = today[: len(sell) + self.topk - len(last)]
+
+        # ---- regime gate -----------------------------------------------------
+        if buy:
+            regime = self._regime_for(buy, pred_start_time, pred_end_time)
+            gated = [c for c in buy if regime.get(c, np.nan) >= self.regime_threshold]
+        else:
+            gated = []
+
+        for code in current_stock_list:
+            if not self.trade_exchange.is_stock_tradable(
+                stock_id=code,
+                start_time=trade_start_time,
+                end_time=trade_end_time,
+                direction=None if self.forbid_all_trade_at_limit else OrderDir.SELL,
+            ):
+                continue
+            if code in sell:
+                time_per_step = self.trade_calendar.get_freq()
+                if current_temp.get_stock_count(code, bar=time_per_step) < self.hold_thresh:
+                    continue
+                sell_amount = current_temp.get_stock_amount(code=code)
+                sell_order = Order(
+                    stock_id=code,
+                    amount=sell_amount,
+                    start_time=trade_start_time,
+                    end_time=trade_end_time,
+                    direction=Order.SELL,
+                )
+                if self.trade_exchange.check_order(sell_order):
+                    sell_order_list.append(sell_order)
+                    trade_val, trade_cost, trade_price = self.trade_exchange.deal_order(
+                        sell_order, position=current_temp
+                    )
+                    cash += trade_val - trade_cost
+
+        if len(gated) == 0:
+            return TradeDecisionWO(sell_order_list, self)
+
+        value = cash * self.risk_degree / len(gated)
+        for code in gated:
+            if not self.trade_exchange.is_stock_tradable(
+                stock_id=code,
+                start_time=trade_start_time,
+                end_time=trade_end_time,
+                direction=None if self.forbid_all_trade_at_limit else OrderDir.BUY,
+            ):
+                continue
+            buy_price = self.trade_exchange.get_deal_price(
+                stock_id=code, start_time=trade_start_time, end_time=trade_end_time, direction=OrderDir.BUY
+            )
+            buy_amount = value / buy_price
+            factor = self.trade_exchange.get_factor(
+                stock_id=code, start_time=trade_start_time, end_time=trade_end_time
+            )
+            buy_amount = self.trade_exchange.round_amount_by_trade_unit(buy_amount, factor)
+            buy_order = Order(
+                stock_id=code,
+                amount=buy_amount,
+                start_time=trade_start_time,
+                end_time=trade_end_time,
+                direction=Order.BUY,
+            )
+            buy_order_list.append(buy_order)
+
+        return TradeDecisionWO(sell_order_list + buy_order_list, self)
@@ -0,0 +1,169 @@
+"""Market-neutral top/bottom long-short strategy for cross-sectional signals.
+
+Captures the cross-sectional long-short spread net of costs: buys the top-ranked
+``topk`` names and shorts the bottom-ranked ``topk`` names, equal-weight per
+side, sized to ``risk_degree`` of total value per side. Rebalances daily to the
+current rank (dropout-free: the book converges to the latest top/bottom sets).
+
+The long and short legs use equal notional per side (gross exposure ~2x
+``risk_degree`` of NAV, i.e. approximately market neutral before transaction
+costs). Benchmark neutrality (SPY beta ~ 0) is the secondary sanity metric.
+"""
+
+from __future__ import annotations
+
+from typing import List
+
+import copy
+
+import pandas as pd
+
+from qlib.backtest import Order
+from qlib.backtest.decision import OrderDir, TradeDecisionWO
+from qlib.contrib.strategy.signal_strategy import BaseSignalStrategy
+
+__all__ = ["TopBottomDropoutStrategy"]
+
+DEFAULT_SHORT_LEG = True
+DEFAULT_REBALANCE_DAILY = True
+
+
+class TopBottomDropoutStrategy(BaseSignalStrategy):
+    """Long top-k / short bottom-k equal-weight market-neutral book.
+
+    Parameters
+    ----------
+    topk : number of names on each side (long top-k and short bottom-k).
+    short_leg : whether to open the short side (if False, long-only topk).
+    rebalance_daily : if True rebalance to current rank every day; else keep
+        positions and only refresh on score changes (dropout-style).
+    risk_degree : fraction of total value deployed per side.
+    """
+
+    def __init__(
+        self,
+        *,
+        topk: int = 10,
+        short_leg: bool = DEFAULT_SHORT_LEG,
+        rebalance_daily: bool = DEFAULT_REBALANCE_DAILY,
+        **kwargs,
+    ):
+        super().__init__(**kwargs)
+        self.topk = topk
+        self.short_leg = short_leg
+        self.rebalance_daily = rebalance_daily
+        self._prev_longs = set()
+        self._prev_shorts = set()
+
+    def generate_trade_decision(self, execute_result=None):
+        trade_step = self.trade_calendar.get_trade_step()
+        trade_start_time, trade_end_time = self.trade_calendar.get_step_time(trade_step)
+        pred_start_time, pred_end_time = self.trade_calendar.get_step_time(trade_step, shift=1)
+        pred_score = self.signal.get_signal(start_time=pred_start_time, end_time=pred_end_time)
+        if isinstance(pred_score, pd.DataFrame):
+            pred_score = pred_score.iloc[:, 0]
+        if pred_score is None or len(pred_score) == 0:
+            return TradeDecisionWO([], self)
+
+        # rank all names; topk longs and topk shorts
+        ranked = pred_score.sort_values(ascending=False)
+        longs = list(ranked.index[: self.topk])
+        shorts = list(ranked.index[-self.topk :]) if self.short_leg else []
+
+        current_temp: "object" = copy.deepcopy(self.trade_position)
+        current_codes = set(current_temp.get_stock_list())
+        holdings = {c: current_temp for c in current_codes if abs(current_temp.get_stock_amount(c)) > 1e-6}
+
+        sell_orders: List[Order] = []
+        buy_orders: List[Order] = []
+
+        def _tradable(code, direction):
+            try:
+                return self.trade_exchange.is_stock_tradable(
+                    stock_id=code, start_time=trade_start_time, end_time=trade_end_time, direction=direction
+                )
+            except TypeError:
+                return self.trade_exchange.is_stock_tradable(
+                    stock_id=code, start_time=trade_start_time, end_time=trade_end_time
+                )
+
+        # determine target set (long/short)
+        target_longs = set(longs)
+        target_shorts = set(shorts)
+
+        # close positions not in the target book
+        for code in list(holdings):
+            if code in target_longs or code in target_shorts:
+                continue
+            amt = abs(current_temp.get_stock_amount(code))
+            o = Order(
+                stock_id=code,
+                amount=amt,
+                start_time=trade_start_time,
+                end_time=trade_end_time,
+                direction=Order.SELL if code in target_longs else Order.SELL,
+            )
+            if self.trade_exchange.check_order(o):
+                sell_orders.append(o)
+                self.trade_exchange.deal_order(o, position=current_temp)
+
+        # equal-weight notional per side
+        total_value = current_temp.get_cash()
+        for code, pos in holdings.items():
+            if code in target_longs or code in target_shorts:
+                mark = self.trade_exchange.get_deal_price(
+                    stock_id=code, start_time=trade_start_time, end_time=trade_end_time, direction=Order.SELL
+                )
+                if mark is not None and mark == mark:
+                    total_value += abs(current_temp.get_stock_amount(code)) * mark
+
+        side_notional = total_value * self.risk_degree / max(1, self.topk)
+
+        for code in longs:
+            if code in holdings and abs(current_temp.get_stock_amount(code)) > 1e-6:
+                continue
+            px = self.trade_exchange.get_deal_price(
+                stock_id=code, start_time=trade_start_time, end_time=trade_end_time, direction=Order.BUY
+            )
+            if px is None or px != px or px <= 0:
+                continue
+            amount = side_notional / px
+            factor = self.trade_exchange.get_factor(
+                stock_id=code, start_time=trade_start_time, end_time=trade_end_time
+            )
+            amount = self.trade_exchange.round_amount_by_trade_unit(amount, factor)
+            o = Order(
+                stock_id=code,
+                amount=amount,
+                start_time=trade_start_time,
+                end_time=trade_end_time,
+                direction=Order.BUY,
+            )
+            if self.trade_exchange.check_order(o):
+                buy_orders.append(o)
+
+        if self.short_leg:
+            for code in shorts:
+                if code in holdings and abs(current_temp.get_stock_amount(code)) > 1e-6:
+                    continue
+                px = self.trade_exchange.get_deal_price(
+                    stock_id=code, start_time=trade_start_time, end_time=trade_end_time, direction=Order.SELL
+                )
+                if px is None or px != px or px <= 0:
+                    continue
+                amount = side_notional / px
+                factor = self.trade_exchange.get_factor(
+                    stock_id=code, start_time=trade_start_time, end_time=trade_end_time
+                )
+                amount = self.trade_exchange.round_amount_by_trade_unit(amount, factor)
+                o = Order(
+                    stock_id=code,
+                    amount=amount,
+                    start_time=trade_start_time,
+                    end_time=trade_end_time,
+                    direction=Order.SELL,
+                )
+                if self.trade_exchange.check_order(o):
+                    sell_orders.append(o)
+
+        return TradeDecisionWO(sell_orders + buy_orders, self)
@@ -0,0 +1,202 @@
+"""Weekly-rebalance TopkDropout strategy.
+
+Turnover-reduction variant of ``qlib.contrib.strategy.signal_strategy.TopkDropoutStrategy``:
+the topk/n_drop selection and sizing are identical to the reference, but the
+target book is recomputed only on the first trading day of each ISO week; on the
+other days the strategy issues NO orders (holds the book untouched).
+
+The weekly cadence is derived from the qlib trade calendar: a rebalance happens
+when the current trade step's date belongs to a different ISO ``(year, week)``
+than the previous trade step. ``hold_band_pct`` (default 0) optionally skips
+tiny rebalances: when a name's existing position differs from the new target by
+less than this fraction, no order is generated for it.
+"""
+
+from __future__ import annotations
+
+from typing import List
+
+import numpy as np
+import pandas as pd
+
+from qlib.backtest import Order
+from qlib.backtest.decision import OrderDir, TradeDecisionWO
+from qlib.contrib.strategy.signal_strategy import TopkDropoutStrategy
+
+__all__ = ["WeeklyRebalanceDropoutStrategy"]
+
+DEFAULT_HOLD_BAND_PCT = 0.0
+
+
+class WeeklyRebalanceDropoutStrategy(TopkDropoutStrategy):
+    """TopkDropout rebalanced once per ISO week; holds otherwise.
+
+    Parameters
+    ----------
+    topk, n_drop, method_sell, method_buy, hold_thresh, only_tradable,
+    forbid_all_trade_at_limit : same as ``TopkDropoutStrategy``.
+    hold_band_pct : skip order for a name whose deviation from target weight is
+        below this fraction of the target (no-trade buffer band).
+    """
+
+    def __init__(self, *, topk, n_drop, hold_band_pct: float = DEFAULT_HOLD_BAND_PCT, **kwargs):
+        super().__init__(topk=topk, n_drop=n_drop, **kwargs)
+        self.hold_band_pct = hold_band_pct
+
+    @staticmethod
+    def _iso_week(ts) -> tuple:
+        return (ts.year, ts.week)
+
+    def generate_trade_decision(self, execute_result=None):
+        import copy
+
+        trade_step = self.trade_calendar.get_trade_step()
+        trade_start_time, trade_end_time = self.trade_calendar.get_step_time(trade_step)
+
+        cur_week = self._iso_week(trade_start_time)
+        prev_week = getattr(self, "_last_week", None)
+        self._last_week = cur_week
+
+        if prev_week is not None and prev_week == cur_week:
+            # not the first trading day of this ISO week -> hold
+            return TradeDecisionWO([], self)
+
+        pred_start_time, pred_end_time = self.trade_calendar.get_step_time(trade_step, shift=1)
+        pred_score = self.signal.get_signal(start_time=pred_start_time, end_time=pred_end_time)
+        if isinstance(pred_score, pd.DataFrame):
+            pred_score = pred_score.iloc[:, 0]
+        if pred_score is None:
+            return TradeDecisionWO([], self)
+
+        if self.only_tradable:
+
+            def get_first_n(li, n, reverse=False):
+                cur_n = 0
+                res = []
+                for si in reversed(li) if reverse else li:
+                    if self.trade_exchange.is_stock_tradable(
+                        stock_id=si, start_time=trade_start_time, end_time=trade_end_time
+                    ):
+                        res.append(si)
+                        cur_n += 1
+                        if cur_n >= n:
+                            break
+                return res[::-1] if reverse else res
+
+            def get_last_n(li, n):
+                return get_first_n(li, n, reverse=True)
+
+            def filter_stock(li):
+                return [
+                    si
+                    for si in li
+                    if self.trade_exchange.is_stock_tradable(
+                        stock_id=si, start_time=trade_start_time, end_time=trade_end_time
+                    )
+                ]
+
+        else:
+
+            def get_first_n(li, n):
+                return list(li)[:n]
+
+            def get_last_n(li, n):
+                return list(li)[-n:]
+
+            def filter_stock(li):
+                return li
+
+        current_temp: "object" = copy.deepcopy(self.trade_position)
+        sell_order_list: List[Order] = []
+        buy_order_list: List[Order] = []
+        cash = current_temp.get_cash()
+        current_stock_list = current_temp.get_stock_list()
+        last = pred_score.reindex(current_stock_list).sort_values(ascending=False).index
+
+        if self.method_buy == "top":
+            today = get_first_n(
+                pred_score[~pred_score.index.isin(last)].sort_values(ascending=False).index,
+                self.n_drop + self.topk - len(last),
+            )
+        elif self.method_buy == "random":
+            topk_candi = get_first_n(pred_score.sort_values(ascending=False).index, self.topk)
+            candi = list(filter(lambda x: x not in last, topk_candi))
+            n = self.n_drop + self.topk - len(last)
+            try:
+                today = np.random.choice(candi, n, replace=False)
+            except ValueError:
+                today = candi
+        else:
+            raise NotImplementedError(f"This type of input is not supported")
+
+        comb = pred_score.reindex(last.union(pd.Index(today))).sort_values(ascending=False).index
+
+        if self.method_sell == "bottom":
+            sell = last[last.isin(get_last_n(comb, self.n_drop))]
+        elif self.method_sell == "random":
+            candi = filter_stock(last)
+            try:
+                sell = pd.Index(np.random.choice(candi, self.n_drop, replace=False) if len(last) else [])
+            except ValueError:
+                sell = candi
+        else:
+            raise NotImplementedError(f"This type of input is not supported")
+
+        buy = today[: len(sell) + self.topk - len(last)]
+        for code in current_stock_list:
+            if not self.trade_exchange.is_stock_tradable(
+                stock_id=code,
+                start_time=trade_start_time,
+                end_time=trade_end_time,
+                direction=None if self.forbid_all_trade_at_limit else OrderDir.SELL,
+            ):
+                continue
+            if code in sell:
+                time_per_step = self.trade_calendar.get_freq()
+                if current_temp.get_stock_count(code, bar=time_per_step) < self.hold_thresh:
+                    continue
+                sell_amount = current_temp.get_stock_amount(code=code)
+                sell_order = Order(
+                    stock_id=code,
+                    amount=sell_amount,
+                    start_time=trade_start_time,
+                    end_time=trade_end_time,
+                    direction=Order.SELL,
+                )
+                if self.trade_exchange.check_order(sell_order):
+                    sell_order_list.append(sell_order)
+                    trade_val, trade_cost, trade_price = self.trade_exchange.deal_order(
+                        sell_order, position=current_temp
+                    )
+                    cash += trade_val - trade_cost
+
+        if len(buy) == 0:
+            return TradeDecisionWO(sell_order_list, self)
+
+        value = cash * self.risk_degree / len(buy)
+        for code in buy:
+            if not self.trade_exchange.is_stock_tradable(
+                stock_id=code,
+                start_time=trade_start_time,
+                end_time=trade_end_time,
+                direction=None if self.forbid_all_trade_at_limit else OrderDir.BUY,
+            ):
+                continue
+            buy_price = self.trade_exchange.get_deal_price(
+                stock_id=code, start_time=trade_start_time, end_time=trade_end_time, direction=OrderDir.BUY
+            )
+            buy_amount = value / buy_price
+            factor = self.trade_exchange.get_factor(
+                stock_id=code, start_time=trade_start_time, end_time=trade_end_time
+            )
+            buy_amount = self.trade_exchange.round_amount_by_trade_unit(buy_amount, factor)
+            buy_order = Order(
+                stock_id=code,
+                amount=buy_amount,
+                start_time=trade_start_time,
+                end_time=trade_end_time,
+                direction=Order.BUY,
+            )
+            buy_order_list.append(buy_order)
+
+        return TradeDecisionWO(sell_order_list + buy_order_list, self)