问题描述:笔者在量化交易投资研究中,保存了一个机器学习训练文件,发现在我的策略中进行读取该文件的时候,会报错找不到文件。
研究平台示例,保存了笔者的一个神经网络训练文件
import pickle
# 保存模型
with open('model.txt', 'wb') as f:
pickle.dump(BP, f)
文件已显示保存成功,并在研究平台目录下生成此文件
这个时候,笔者 打算到我的策略中,读取研究平台保存的文件,使用如下代码读取
# 读取模型
with open('model.txt', 'rb') as f:
model = pickle.load(f)
model.predict(x_train)
报错,找不到文件,笔者想了很久,测试了很多方法,发现依然无法读取文件
那索性查看下平台api看看还有什么读取文件和保存文件的方式。发现平台仅支持它提供的已有的几种文件读取方式,别的方式,一定是会找不到文件而导致报错,于是,我找到了另一个函数:
get_file函数
body = get_file('model.txt')
context.model = pickle.load(BytesIO(body))
通过以上函数,解决找不到文件问题