Implement Trader V4 training artifact pipeline

This commit is contained in:
Codex
2026-06-27 16:15:23 +08:00
parent dad6b831b4
commit e58e4a5572
113 changed files with 7959 additions and 477 deletions
+23
View File
@@ -0,0 +1,23 @@
from __future__ import annotations
import argparse
import _bootstrap # noqa: F401
from trader_training.io_utils import add_common_args, setup_logging
from trader_training.replay import write_audit_outputs
def main() -> None:
parser = argparse.ArgumentParser()
add_common_args(parser)
parser.add_argument("--symbol", default="BTC-USDT-PERP")
parser.add_argument("--start-date")
parser.add_argument("--end-date")
parser.add_argument("--min-ready-days", type=int, default=250)
args = parser.parse_args()
setup_logging()
write_audit_outputs(args)
if __name__ == "__main__":
main()
+26
View File
@@ -0,0 +1,26 @@
from __future__ import annotations
import argparse
from pathlib import Path
import _bootstrap # noqa: F401
from trader_training.io_utils import add_common_args, setup_logging
from trader_training.replay import build_replay_1m
def main() -> None:
parser = argparse.ArgumentParser()
add_common_args(parser)
parser.add_argument("--raw-root", type=Path)
parser.add_argument("--symbol", default="BTC-USDT-PERP")
parser.add_argument("--start-date")
parser.add_argument("--end-date")
parser.add_argument("--min-minutes-per-day", type=int, default=1400)
parser.add_argument("--min-replay-ready-days", type=int, default=250)
args = parser.parse_args()
setup_logging()
build_replay_1m(args)
if __name__ == "__main__":
main()
+31
View File
@@ -0,0 +1,31 @@
from __future__ import annotations
import argparse
from pathlib import Path
import _bootstrap # noqa: F401
from trader_training.io_utils import add_common_args, setup_logging
from trader_training.replay import build_splits
def main() -> None:
parser = argparse.ArgumentParser()
add_common_args(parser)
parser.add_argument("--replay-path", type=Path)
parser.add_argument("--fit-inner-start", default="2025-06-20")
parser.add_argument("--fit-inner-end", default="2026-01-15")
parser.add_argument("--tune-inner-start", default="2026-01-16")
parser.add_argument("--tune-inner-end", default="2026-02-28")
parser.add_argument("--validation-locked-start", default="2026-03-01")
parser.add_argument("--validation-locked-end", default="2026-04-30")
parser.add_argument("--latest-stress-start", default="2026-05-01")
parser.add_argument("--latest-stress-end", default="2026-06-19")
parser.add_argument("--gap-minutes", type=int, default=60)
parser.add_argument("--fold-count", type=int, default=3)
args = parser.parse_args()
setup_logging()
build_splits(args)
if __name__ == "__main__":
main()
@@ -0,0 +1,23 @@
from __future__ import annotations
import argparse
from pathlib import Path
import _bootstrap # noqa: F401
from trader_training.features import build_feature_frame
from trader_training.io_utils import add_common_args, setup_logging
def main() -> None:
parser = argparse.ArgumentParser()
add_common_args(parser)
parser.add_argument("--replay-path", type=Path)
parser.add_argument("--split-manifest-path", type=Path)
parser.add_argument("--allow-incomplete-days", action="store_true")
args = parser.parse_args()
setup_logging()
build_feature_frame(args)
if __name__ == "__main__":
main()
@@ -0,0 +1,23 @@
from __future__ import annotations
import argparse
from pathlib import Path
import _bootstrap # noqa: F401
from trader_training.io_utils import add_common_args, setup_logging
from trader_training.labels import write_price_plan_context
def main() -> None:
parser = argparse.ArgumentParser()
add_common_args(parser)
parser.add_argument("--label-config-path", type=Path)
parser.add_argument("--cost-config-path", type=Path)
parser.add_argument("--price-plan-id", default="btc-p0-plan-45m")
args = parser.parse_args()
setup_logging()
write_price_plan_context(args)
if __name__ == "__main__":
main()
@@ -0,0 +1,23 @@
from __future__ import annotations
import argparse
from pathlib import Path
import _bootstrap # noqa: F401
from trader_training.io_utils import add_common_args, setup_logging
from trader_training.labels import build_direction_labels
def main() -> None:
parser = argparse.ArgumentParser()
add_common_args(parser)
parser.add_argument("--feature-path", type=Path)
parser.add_argument("--replay-path", type=Path)
parser.add_argument("--label-config-path", type=Path)
args = parser.parse_args()
setup_logging()
build_direction_labels(args)
if __name__ == "__main__":
main()
+25
View File
@@ -0,0 +1,25 @@
from __future__ import annotations
import argparse
from pathlib import Path
import _bootstrap # noqa: F401
from trader_training.io_utils import add_common_args, setup_logging
from trader_training.labels import build_entry_labels
def main() -> None:
parser = argparse.ArgumentParser()
add_common_args(parser)
parser.add_argument("--feature-path", type=Path)
parser.add_argument("--replay-path", type=Path)
parser.add_argument("--label-config-path", type=Path)
parser.add_argument("--cost-config-path", type=Path)
parser.add_argument("--price-plan-context-path", type=Path)
args = parser.parse_args()
setup_logging()
build_entry_labels(args)
if __name__ == "__main__":
main()
@@ -0,0 +1,21 @@
from __future__ import annotations
import argparse
from pathlib import Path
import _bootstrap # noqa: F401
from trader_training.io_utils import add_common_args, setup_logging
from trader_training.labels import build_position_state_samples
def main() -> None:
parser = argparse.ArgumentParser()
add_common_args(parser)
parser.add_argument("--entry-label-path", type=Path)
args = parser.parse_args()
setup_logging()
build_position_state_samples(args)
if __name__ == "__main__":
main()
@@ -0,0 +1,25 @@
from __future__ import annotations
import argparse
from pathlib import Path
import _bootstrap # noqa: F401
from trader_training.io_utils import add_common_args, setup_logging
from trader_training.labels import build_continue_exit_risk_labels
def main() -> None:
parser = argparse.ArgumentParser()
add_common_args(parser)
parser.add_argument("--feature-path", type=Path)
parser.add_argument("--replay-path", type=Path)
parser.add_argument("--label-config-path", type=Path)
parser.add_argument("--cost-config-path", type=Path)
parser.add_argument("--price-plan-context-path", type=Path)
args = parser.parse_args()
setup_logging()
build_continue_exit_risk_labels(args)
if __name__ == "__main__":
main()
@@ -0,0 +1,26 @@
from __future__ import annotations
import argparse
from pathlib import Path
import _bootstrap # noqa: F401
from trader_training.datasets import build_train_datasets
from trader_training.io_utils import add_common_args, setup_logging
def main() -> None:
parser = argparse.ArgumentParser()
add_common_args(parser)
parser.add_argument("--feature-path", type=Path)
parser.add_argument("--direction-label-path", type=Path)
parser.add_argument("--entry-label-path", type=Path)
parser.add_argument("--continue-label-path", type=Path)
parser.add_argument("--exit-label-path", type=Path)
parser.add_argument("--risk-label-path", type=Path)
args = parser.parse_args()
setup_logging()
build_train_datasets(args)
if __name__ == "__main__":
main()
+20
View File
@@ -0,0 +1,20 @@
from __future__ import annotations
import argparse
import _bootstrap # noqa: F401
from trader_training.io_utils import add_common_args, setup_logging
from trader_training.training import train_small_models
def main() -> None:
parser = argparse.ArgumentParser()
add_common_args(parser)
parser.add_argument("--max-rows", type=int, default=0)
args = parser.parse_args()
setup_logging()
train_small_models(args)
if __name__ == "__main__":
main()
+19
View File
@@ -0,0 +1,19 @@
from __future__ import annotations
import argparse
import _bootstrap # noqa: F401
from trader_training.io_utils import add_common_args, setup_logging
from trader_training.training import build_calibrators
def main() -> None:
parser = argparse.ArgumentParser()
add_common_args(parser)
args = parser.parse_args()
setup_logging()
build_calibrators(args)
if __name__ == "__main__":
main()
@@ -0,0 +1,19 @@
from __future__ import annotations
import argparse
import _bootstrap # noqa: F401
from trader_training.io_utils import add_common_args, setup_logging
from trader_training.pm import search_pm_thresholds
def main() -> None:
parser = argparse.ArgumentParser()
add_common_args(parser)
args = parser.parse_args()
setup_logging()
search_pm_thresholds(args)
if __name__ == "__main__":
main()
@@ -0,0 +1,19 @@
from __future__ import annotations
import argparse
import _bootstrap # noqa: F401
from trader_training.io_utils import add_common_args, setup_logging
from trader_training.pm import integrated_backtest
def main() -> None:
parser = argparse.ArgumentParser()
add_common_args(parser)
args = parser.parse_args()
setup_logging()
integrated_backtest(args)
if __name__ == "__main__":
main()
@@ -0,0 +1,21 @@
from __future__ import annotations
import argparse
from pathlib import Path
import _bootstrap # noqa: F401
from trader_training.exporter import export_artifact_bundle
from trader_training.io_utils import add_common_args, setup_logging
def main() -> None:
parser = argparse.ArgumentParser()
add_common_args(parser)
parser.add_argument("--export-root", type=Path)
args = parser.parse_args()
setup_logging()
export_artifact_bundle(args)
if __name__ == "__main__":
main()
@@ -0,0 +1,22 @@
from __future__ import annotations
import argparse
from pathlib import Path
import _bootstrap # noqa: F401
from trader_training.io_utils import setup_logging
from trader_training.validator import validate_artifact_bundle
def main() -> None:
parser = argparse.ArgumentParser()
parser.add_argument("--artifact-root", type=Path, required=True)
parser.add_argument("--require-active", action="store_true")
parser.add_argument("--run-onnx", action="store_true")
args = parser.parse_args()
setup_logging()
validate_artifact_bundle(args)
if __name__ == "__main__":
main()
@@ -0,0 +1,21 @@
from __future__ import annotations
import argparse
from pathlib import Path
import _bootstrap # noqa: F401
from trader_training.io_utils import setup_logging
from trader_training.promote import promote_artifact_bundle
def main() -> None:
parser = argparse.ArgumentParser()
parser.add_argument("--artifact-root", type=Path, required=True)
parser.add_argument("--reason", required=True)
args = parser.parse_args()
setup_logging()
promote_artifact_bundle(args)
if __name__ == "__main__":
main()
+8
View File
@@ -0,0 +1,8 @@
from __future__ import annotations
import sys
from pathlib import Path
TRAINING_ROOT = Path(__file__).resolve().parents[1]
if str(TRAINING_ROOT) not in sys.path:
sys.path.insert(0, str(TRAINING_ROOT))