train_estimator

czsc.ai.train_estimator(estimator: Pipeline, df: DataFrame, x_cols: List, y_col: AnyStr, train_days: int, valid_days: int, method='rolling')[source]

训练模型

Parameters:
  • estimator – 使用 Pipeline 组合成模型

  • df – 数据对象

  • x_cols – 特征名序列

  • y_col – 真实值列

  • train_days – 训练集时间跨度

  • valid_days – 验证集时间跨度

  • method – 时间窗口滚动方法,rolling 滑动窗口,expanding 扩张窗口

Returns: