Python数据类型的保存和载入方法

一、集合和字典

import numpy as np

保存：

np.save('filename.npy', data)

载入：

data = np.load('filename.npy', allow_pickle=True).item()

data：集合或字典类型

二、keras模型

from keras.models import load_model

保存：

model.save('filename.h5')

载入：

model = load_model('filename.h5')

model：keras模型

三、数组

import numpy as np

保存：

np.save('filename.npy', data)

载入：

data = np.load('filename.npy', allow_pickle=True)

data：数组

四、sklearn模型

import joblib

保存：

joblib.dump(model, 'filename.pkl')

载入：

model = joblib.load('filename.pkl')

model：sklearn模型

五、pytorch模型

import torch

保存：

torch.save(model, 'filename.pkl')

载入：

model = torch.load('filename.pkl')

model：pytorch模型