Sfoglia il codice sorgente

更改SHAP计算batch_size从500到200

Sherlock1011 4 mesi fa
parent
commit
b04db1e8bb
1 ha cambiato i file con 1 aggiunte e 1 eliminazioni
  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()