我有一个数据集,其中包含
列从这个数据集中,我想训练一个数据模型,该模型给出了许多项目,权重和类别可以预测持续时间(结束时间-开始时间)。
如何将标签列转换或设置为持续时间,以便获得EstimatorChain
并可以与从CSV加载的Fit
一起调用IDataView
?
答案 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属性。