Home Explore Help
Sign In
huanghongbo
/
BrandCultivation
1
0
Fork 0
Files Issues 0 Pull Requests 0 Wiki
Browse Source

更改SHAP计算batch_size从500到200

Sherlock1011 4 months ago
parent
43b1358a13
commit
b04db1e8bb
1 changed files with 1 additions and 1 deletions
Split View Show Diff Stats
  1. 1 1
      models/rank/gbdt_lr_inference.py

+ 1 - 1
models/rank/gbdt_lr_inference.py
View File 

@@ -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()