python 使用joblib在内存中存储和检索对象

Posted 2021-05-09

from tempfile import mkdtemp
savedir = mkdtemp()
import os
filename = os.path.join(savedir, 'test.pkl')

import numpy as np
to_persist = [('a', [1, 2, 3]), ('b', np.arange(10))]

import joblib
joblib.dump(to_persist, filename)  
joblib.load(filename)