求教:回测引擎中如何取到每日的preditions的值


(youke) #1

如题,

针对训练好的模型并预测

预测

predictions = model.predict(test_x)

而在后面的回测引擎中,如何取到每日的preditions的值?

回测引擎:每日数据处理函数,每天执行一次

def handle_data(context, data):
current_dt = data.current_dt.strftime(’%Y-%m-%d’)
。。。


(iQuant) #2

按道理来说,predictions是有date(日期)列的,因此,你可以:

predictions_stocks = predictions.set_index('date').ix[current_dt]

此外,你可以借鉴ai量化模板策略相关代码:

 ranker_prediction = context.ranker_prediction[
        context.ranker_prediction.date == data.current_dt.strftime('%Y-%m-%d')]