有办法减少PyCaret模型的泡菜文件的大小吗？

Question

我试图压缩一个表示PyCaret模型的泡菜文件。

import joblib
joblib.dump('my_file.pkl',  'new_file.pkl.z',compress=3)

以上代码不起作用。

如何减少PyCaret模型的泡菜文件的大小？

Answer 1

不需要自己调用joblib.dump函数，您应该在加载的模型上直接调用PyCaret保存函数，如下所示：

from pycaret.regression import load_model, save_model

your_model = load_model('my_file.pkl')
save_model(your_model, f'my_file.pkl', **{"compress":3})

上面的例子是回归模型。

供参考: Pycaret调用传递任何给定kwargs的jolib.dump函数这里。