Compressor/models/vibration/infer_and_combine.py

#!/usr/bin/env python3
"""Generate the vibration 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.vibration import features_vibration


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


if __name__ == "__main__":
    main()