python保存字典和读取字典pickle

import pickle
import numpy as np


def save_obj(obj, name):
    with open(name + '.pkl', 'wb') as f:
        pickle.dump(obj, f, pickle.HIGHEST_PROTOCOL)


def load_obj(name):
    with open(name + '.pkl', 'rb') as f:
        return pickle.load(f)


a = load_obj('logistic_regression_res')
W = np.array(a['W'])
b = np.array(a['b'])
print(W)
print(b)