在lightGBM mac上加载M1模型时发生分割故障
在lightGBM mac上加载M1模型时发生分割故障
提问于 2022-03-05 07:52:18
当我试图加载LightGBM模型时，分割错误就会发生。我已经尝试过泡菜(包括协议更改)、云端泡菜、load和lightGBM save_model，但是，我无法加载这些模型。
保存模式：
model = lgb.LGBMRegressor(**kargs)
model.fit(self.X_train, self.y_train)
with open(self.output_dir + 'models.pkl', "wb") as f:
pickle.dump(model, f, protocol=4)
with open(self.output_dir + 'modelcl.pkl', "wb") as f:
f.write(cloudpickle.dumps(model))
pickle.dump(model, open(self.output_dir +
'model.pkl', 'wb'), protocol=4)
joblib.dump(model, f'{self.output_dir}model.joblib')
joblib.dump(model, f'{self.output_dir}model.xz', compress=True)
model.booster_.save_model(f'{self.output_dir}model.txt')负荷模型：
joblib.load('path/model.xz')
joblib.load('path/model.joblib')
pickle.load(open('path/models.pkl', 'rb'))
pickle.load(open('path/modelcl.pkl', 'rb'))
pickle.load(open('path/model.pkl', 'rb'))
model = lgb.Booster(model_file='path/model.txt')环境: M1 Mac蒙特雷(12.0.1)Python3.8.12 (conda)，joblib==1.1.0，lightgbm==3.3.2。
你知道这是怎么回事吗？我认为我应该尝试保存权重，而不是保存整个模型，但不知道如何在LightGBM上。
回答 1
Stack Overflow用户
发布于 2022-03-05 10:40:18
通过从conda conda install -c conda-forge lightgbm安装lightgbm而不是使用pip install lightgbm解决了这个问题。可能不建议从pip安装lightgbm。
已安装的软件包是conda-forge/osx-arm64::lightgbm-3.3.2-py38hb2dfaa5_0
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/71360416
复制相关文章
相似问题
腾讯云开发者
