如何使用投影代替原始列

时间:2019-04-30 12:16:01

标签: projection ml.net

我有一个数据集,其中包含

  • 项目数(整数值)
  • 物品的重量(分数)
  • 产品类别
  • 开始时间
  • 结束时间

从这个数据集中,我想训练一个数据模型,该模型给出了许多项目,权重和类别可以预测持续时间(结束时间-开始时间)。

如何将标签列转换或设置为持续时间,以便获得EstimatorChain并可以与从CSV加载的Fit一起调用IDataView

1 个答案:

答案 0 :(得分:2)

您可以使用ML.NET CustomMapping功能来计算持续时间,然后调用该Label。 This contains an example of how to use it.

Action<Data, Data> mapping =
            (input, output) => output.Label= input.End - Input.Start;

其中,Data是您的数据模型,除了其他属性外,它还包含Label属性。