添加耗时统计装饰器工具

2025-10-26 00:51:22 +08:00
parent 332284293f
commit 982d0fda11
2 changed files with 24 additions and 3 deletions
--- a/common/bet_tools.py
+++ b/common/bet_tools.py
@@ -250,14 +250,14 @@ def compute_metrics(
    beta = float(clf.coef_[0][0])

    return {
-        "logloss": logloss,
+        "n_samples": len(df),
        "brier": brier,
-        "ece": ece,
+        "logloss": logloss,
        "accuracy": acc,
        "reg_alpha": alpha,
        "reg_beta": beta,
+        "ece": ece,
        # 'ece_bins': bin_stats,
-        "n_samples": int(total),
    }