【问题标题】:Is there a way to reduce down the size of the pickle file for PyCaret models?有没有办法减少 PyCaret 模型的 pickle 文件的大小?
【发布时间】:2022-08-23 19:04:32
【问题描述】:

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

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

上面的代码不起作用。

如何减小 PyCaret 模型的 pickle 文件的大小?

    标签: pickle pycaret


    【解决方案1】:

    不需要自己调用joblib.dump函数,直接在加载的模型上调用PyCaret save函数如下:

    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})
    

    上面的例子是一个回归模型

    【讨论】:

      最近更新 更多