【第一步】自定义模型集成
- 简介
- 自定义模型类
- 配置文件
- 模型测试
简介
Qlib 的模型库包括 LightGBM、MLP、LSTM 等模型。这些模型都是预测模型的例子。除了 Qlib 提供的默认模型外，用户还可以将他们自己的自定义模型集成到 Qlib 中。
用户可以按照以下步骤集成他们自己的自定义模型。
- 定义一个自定义模型类，它应该是 qlib.model.base.Model 的子类。
- 编写一个描述自定义模型的路径和参数的配置文件。
- 测试自定义模型。
自定义模型类
自定义模型需要继承 qlib.model.base.Model 并重写其中的方法。
- 重写 __init__方法
- Qlib 将初始化的参数传递给__init__ 方法。
- 配置中的模型超参数必须与__init__ 方法中定义的一致。
- 代码示例：在以下示例中，配置文件中的模型超参数应该包含例如 loss:mse 这样的参数。
def __init__(self, loss
)
