如何使用投影而不是原始列

Question

我有一个包含

列的数据集

• 项目数（整数值）
• 物品重量（分数值）
• 一个产品类别
• 开始时间
• 结束时间

我想从这个数据集中训练一个数据模型，给定一些项目、重量和类别可以预测持续时间（结束时间 - 开始时间）。

如何将我的标签列转换或设置为持续时间，以便我得到一个 EstimatorChain，我可以用我加载的 IDataView 调用 Fit来自 CSV？

Answer 1

您可以使用 ML.NET CustomMapping 功能来计算持续时间，并调用该标签。 This contains an example of how to use it.

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

其中 Data 是包含标签 属性 以及其他属性的数据模型。