fix:评估对齐

2026-03-20 16:52:24 +08:00
parent 77e38fd15b
commit cc85e3807a
2 changed files with 9 additions and 4 deletions
--- a/backend/core/train_model.py
+++ b/backend/core/train_model.py
@@ -56,11 +56,15 @@ def patch_lightgbm_sklearn_compatibility():
        return

    params = inspect.signature(check_X_y).parameters
-    if 'force_all_finite' in params or 'ensure_all_finite' not in params:
+    if 'force_all_finite' in params:
        return

    def wrapped_check_X_y(*args, force_all_finite=None, **kwargs):
-        if force_all_finite is not None and 'ensure_all_finite' not in kwargs:
+        if (
+            force_all_finite is not None
+            and 'ensure_all_finite' in params
+            and 'ensure_all_finite' not in kwargs
+        ):
            kwargs['ensure_all_finite'] = force_all_finite
        return check_X_y(*args, **kwargs)