32 lines
1.0 KiB
Python
32 lines
1.0 KiB
Python
from cross_validation import DataValidator
|
|
import sys
|
|
|
|
def main():
|
|
"""主函数"""
|
|
import argparse
|
|
|
|
parser = argparse.ArgumentParser(
|
|
description='从xtask导出数据中抽取牌面进行EHS验证',
|
|
formatter_class=argparse.RawDescriptionHelpFormatter
|
|
)
|
|
|
|
parser.add_argument('--river-samples', type=int, default=10, help='River样本数 (默认: 10)')
|
|
parser.add_argument('--turn-samples', type=int, default=5, help='Turn样本数 (默认: 5)')
|
|
parser.add_argument('--flop-samples', type=int, default=3, help='Flop样本数 (默认: 3)')
|
|
parser.add_argument('--data-path', type=str, default='ehs_data', help='数据路径 (默认: ehs_data)')
|
|
|
|
args = parser.parse_args()
|
|
|
|
|
|
validator = DataValidator(data_path=args.data_path)
|
|
results = validator.run_full_validation(
|
|
river_samples=args.river_samples,
|
|
turn_samples=args.turn_samples,
|
|
flop_samples=args.flop_samples
|
|
)
|
|
|
|
return 0 if results['overall_success'] else 1
|
|
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(main()) |