5、【第一步】自定义模型集成

简介

Qlib 的模型库包括 LightGBM、MLP、LSTM 等模型。这些模型都是预测模型的例子。除了 Qlib 提供的默认模型外，用户还可以将他们自己的自定义模型集成到 Qlib 中。

用户可以按照以下步骤集成他们自己的自定义模型。

• 定义一个自定义模型类，它应该是 qlib.model.base.Model 的子类。
• 编写一个描述自定义模型的路径和参数的配置文件。
• 测试自定义模型。

自定义模型类

自定义模型需要继承 qlib.model.base.Model 并重写其中的方法。

• 重写 __init__方法
  • Qlib 将初始化的参数传递给__init__ 方法。
  • 配置中的模型超参数必须与__init__ 方法中定义的一致。
  • 代码示例：在以下示例中，配置文件中的模型超参数应该包含例如 loss:mse 这样的参数。

def __init__(self, loss