فهرست منبع

更改SHAP计算batch_size从500到200

Sherlock1011 4 ماه پیش
والد
کامیت
b04db1e8bb
1فایلهای تغییر یافته به همراه1 افزوده شده و 1 حذف شده
  1. 1 1
      models/rank/gbdt_lr_inference.py

+ 1 - 1
models/rank/gbdt_lr_inference.py

@@ -117,7 +117,7 @@ class GbdtLrModel:
         # 获取数据基本信息
         n_samples = len(data)
         n_features = len(data.columns)
-        batch_size = 500  # 可根据内存调整
+        batch_size = 200  # 可根据内存调整
         
         # 创建临时内存映射文件
         # temp_dir = tempfile.mkdtemp()