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