气流：存储机器学习模块

Question

我会在气流上做机器学习的流水线。
（示例）

result = model.fit()

但是DAG文件：（样本.py）每次都刷新。你知道吗

所以我不能储存训练过的模型。你知道吗

我应该如何对待训练有素的模特？你知道吗

或者我应该将经过训练的模型数据存储在外部数据库中？
那就把训练好的模型拿来用吧？你知道吗

Answer 1

将模型存储在pickle文件中，并随时从同一文件中检索模型。你知道吗

import pickle 
result = model.fit()
# save the model to disk
filename = 'my_model.sav'
pickle.dump(model, open(filename, 'wb'))

# load the model from file for later use
reload_model = pickle.load(open(filename, 'rb'))
result_final = reload_model.score(X_test, y_test)
print result_final