Improve nonlinear PM diagnostics

2026-06-28 09:59:36 +08:00
parent 6be4bb976a
commit 0fe3bd864e
4 changed files with 282 additions and 46 deletions
@@ -8,8 +8,21 @@ from trader_training.nonlinear_pm_probe import probe_nonlinear_pm


 def main() -> None:
-    parser = argparse.ArgumentParser(description="Run diagnostic nonlinear Direction + Entry PM probe.")
+    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)