2569718930@qq.com
66 lines
1.8 KiB
Python
66 lines
1.8 KiB
Python
import argparse
|
|
import json
|
|
import os
|
|
import sys
|
|
|
|
PROJECT_ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
|
if PROJECT_ROOT not in sys.path:
|
|
sys.path.insert(0, PROJECT_ROOT)
|
|
|
|
from src.analysis.deb_algorithm import load_history # noqa: E402
|
|
from scripts.fit_probability_calibration import ( # noqa: E402
|
|
_extract_samples,
|
|
_load_json_if_exists,
|
|
)
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description="Export normalized probability calibration training samples.")
|
|
parser.add_argument(
|
|
"--history-file",
|
|
default=os.path.join(PROJECT_ROOT, "data", "daily_records.json"),
|
|
)
|
|
parser.add_argument(
|
|
"--settlement-history",
|
|
default=os.path.join(
|
|
PROJECT_ROOT,
|
|
"artifacts",
|
|
"probability_calibration",
|
|
"settlement_history.json",
|
|
),
|
|
)
|
|
parser.add_argument(
|
|
"--output",
|
|
default=os.path.join(
|
|
PROJECT_ROOT,
|
|
"artifacts",
|
|
"probability_calibration",
|
|
"training_samples.json",
|
|
),
|
|
)
|
|
args = parser.parse_args()
|
|
|
|
history = load_history(args.history_file)
|
|
settlement_history = _load_json_if_exists(args.settlement_history)
|
|
samples, filled_actual_from_history = _extract_samples(
|
|
history,
|
|
settlement_history=settlement_history,
|
|
)
|
|
|
|
output_dir = os.path.dirname(os.path.abspath(args.output))
|
|
if output_dir:
|
|
os.makedirs(output_dir, exist_ok=True)
|
|
payload = {
|
|
"sample_count": len(samples),
|
|
"filled_actual_from_history": filled_actual_from_history,
|
|
"samples": samples,
|
|
}
|
|
with open(args.output, "w", encoding="utf-8") as fh:
|
|
json.dump(payload, fh, ensure_ascii=False, indent=2)
|
|
|
|
print(f"exported {len(samples)} samples to {args.output}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|