cross-validation

task5_main.py

@@ -1,32 +1,8 @@
-from cross_validation import DataValidator
+from cross_validation import cross_validate_main
 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
-
+    return cross_validate_main()
 
 if __name__ == '__main__':
     sys.exit(main())