Qlib如何配置使用模型LSTM进行预测

有同学看了我的Qlib教学视频后,想将视频中的示例机器学习模型LGBModel换成Qlib提供的另一个模型LSTM,但不知道怎么配置。

我看了一下,确实LSTM模型的配置要比LGBModel麻烦。参考Qlib源码中的如下yaml文件:

我们可以在jupyter notebook中写出对应的词典配置,其中改动的是

data_handler_config和task配置段。我已将源码传到腾讯课堂Qlib课程最后一课,欢迎下载。

data_handler_config = {
    "start_time": "2008-01-01",
    "end_time": "2020-08-01",
    "fit_start_time": "2008-01-01",
    "fit_end_time": "2014-12-31",
    "instruments": market,
    "infer_processors":
......

读者可以对照yaml文件和上图中相应配置的对应关系,理解了就可以自己配置任何模型。

猜你喜欢

转载自blog.csdn.net/qtbgo/article/details/113183319