python 保存模型、加载模型 Joblib

目录

python 保存模型、加载模型 Joblib

numpy保存与加载数据：

pytorch 保存与加载

---

python 保存模型、加载模型 Joblib

支持数据类型：字典，列表，numpy。

保存模型：

        saved_data = {"ROI": pts}
        joblib.dump(value=saved_data, filename="config.pkl")

加载模型：

     model1 = joblib.load(filename="config.pkl")

     print(model1)

numpy保存与加载数据：

    np.save('test_a.npy', image)

    a = np.load('test_a.npy')

    diff=abs(a-image)

pytorch 保存与加载

# 保存模型
torch.save(model, 'model.pth\pkl\pt')    #一般形式torch.save(net, PATH) 
# 加载模型
model = torch.load('model.pth\pkl\pt')   #一般形式为model_dict=torch.load(PATH)