Compressor/models/generalized/infer_and_combine.py

#!/usr/bin/env python3
"""Generate the pressure rule/model joint assistance output."""
from __future__ import annotations

import argparse
import sys
from pathlib import Path

sys.path.insert(0, str(Path(__file__).resolve().parents[2]))

from models.common import modeling, paths
from models.generalized import features_pressure


def main() -> None:
    parser = argparse.ArgumentParser(description="生成压力规则与模型的历史联合辅助确认结果")
    parser.add_argument("--datasets-dir", default=str(paths.PRESSURE_DATAS))
    parser.add_argument("--models-dir", default=str(paths.PRESSURE_MODELS))
    parser.add_argument("--output-dir", default=str(paths.PRESSURE_RESULTS))
    args = parser.parse_args()
    result = modeling.infer("pressure", features_pressure, Path(args.datasets_dir), Path(args.models_dir), Path(args.output_dir))
    print(f"pressure 联合辅助结果：{result['rows']:,} 行，输出：{result['output']}")


if __name__ == "__main__":
    main()