Codex
33 lines
1.0 KiB
Python
33 lines
1.0 KiB
Python
from __future__ import annotations
|
|
|
|
import argparse
|
|
|
|
import _bootstrap # noqa: F401
|
|
from trader_training.io_utils import add_common_args, setup_logging
|
|
from trader_training.nonlinear_pm_probe import probe_nonlinear_pm
|
|
|
|
|
|
def main() -> None:
|
|
parser = argparse.ArgumentParser(description="Run diagnostic nonlinear model PM probe.")
|
|
add_common_args(parser)
|
|
parser.add_argument(
|
|
"--probe-mode",
|
|
choices=("direction_entry_tree", "entry_tree_only"),
|
|
default="direction_entry_tree",
|
|
help="诊断模式：同时替换 Direction+Entry，或只替换 Entry、保留当前 Direction 输出。",
|
|
)
|
|
parser.add_argument(
|
|
"--entry-train-filter",
|
|
choices=("direction_label", "side_opportunity"),
|
|
default="direction_label",
|
|
help="树模型 Entry 的训练人群来源。",
|
|
)
|
|
parser.add_argument("--entry-opportunity-bps", type=float, default=40.0)
|
|
args = parser.parse_args()
|
|
setup_logging()
|
|
probe_nonlinear_pm(args)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|