随机森林预测 (v2)

# 随机森林预测 (v2)

定义

M.random_forest_predict.v2(self, model, data, date_col='date', instrument_col='instrument', sort=True)

使用随机森林模型对数据进行预测。如果是回归任务,结果是数值;如果是分类任务,结果是类别和对应类别的概率。

参数:

  • model (字符串) – 模型,参考 随机森林模型 M.random_forest_train。
  • data (DataSource) – 待预测数据集。
  • date_col (str) – 日期列名,如果在表达式中用到切面相关函数时,比如 rank,会用到此列名;默认值是date。
  • instrument_col (str) – 证券代码列名,如果在表达式中用到时间序列相关函数时,比如 shift,会用到此列名;默认值是instrument。
  • sort (bool) – 是否对结果排序;默认值是True。

返回:

  • .predictions: 预测结果

返回类型:

  • Outputs