c# - 如何使用 ML.NET 预测整数值？

Question

我在这里查看 cs 文件:https://www.microsoft.com/net/learn/apps/machine-learning-and-ai/ml-dotnet/get-started/windows一切正常。

现在我想改进这个例子:我想预测一个只有数字的数据集而不是一个数字字符串数据集，例如预测一个七段显示的输出。

这是我 super 简单的数据集，最后一列是我要预测的整数:

1,0,1,1,1,1,1,0
0,0,0,0,0,1,1,1
1,1,1,0,1,1,0,2
1,1,1,0,0,1,1,3
0,1,0,1,0,1,1,4
1,1,1,1,0,0,1,5
1,1,1,1,1,0,1,6
1,0,0,0,0,1,1,7
1,1,1,1,1,1,1,8
1,1,1,1,0,1,1,9

这是我的测试代码:

public class Digit
{
    [Column("0")] public float Up;

    [Column("1")] public float Middle;

    [Column("2")] public float Bottom;

    [Column("3")] public float UpLeft;
    [Column("4")] public float BottomLeft;
    [Column("5")] public float TopRight;
    [Column("6")] public float BottomRight;

    [Column("7")] [ColumnName("DigitValue")]
    public float DigitValue;
}

public class DigitPrediction
{
    [ColumnName("PredictedDigits")] public float PredictedDigits;
}

public PredictDigit()
{
    var pipeline = new LearningPipeline();
    var dataPath = Path.Combine("Segmenti", "segments.txt");
    pipeline.Add(new TextLoader<Digit>(dataPath, false, ","));
    pipeline.Add(new ColumnConcatenator("Label", "DigitValue"));
    pipeline.Add(new ColumnConcatenator("Features", "Up", "Middle", "Bottom", "UpLeft", "BottomLeft", "TopRight", "BottomRight"));
    pipeline.Add(new StochasticDualCoordinateAscentClassifier());
    var model = pipeline.Train<Digit, DigitPrediction>();
    var prediction = model.Predict(new Digit
    {
        Up = 1,
        Middle = 1,
        Bottom = 1,
        UpLeft = 1,
        BottomLeft = 1,
        TopRight = 1,
        BottomRight = 1,
    });

    Console.WriteLine($"Predicted digit is: {prediction.PredictedDigits}");
    Console.ReadLine();
}

如您所见，除了最后一列(“标签”)处理之外，它与提供的示例非常相似，因为我需要预测一个数字而不是一个字符串。我尝试:

pipeline.Add(new ColumnConcatenator("Label", "DigitValue"));

但它不起作用，异常:

Training label column 'Label' type is not valid for multi-class: Vec<R4, 1>. Type must be R4 or R8.

我确定我错过了一些东西，但实际上我在互联网上找不到任何可以帮助我解决这个问题的东西。

更新

我发现数据集必须有一个像这样的Label列:

[Column("7")] [ColumnName("Label")] public float Label;

和 DigitPrediction 一个 Score 列，例如:

public class DigitPrediction
{
    [ColumnName("Score")] public float[] Score;
}

现在系统“工作”了，我得到了 prediction.Score 一个 Single[] 值，其中与较高值关联的索引是预测值。 这是正确的方法吗？

更新 2 - 完整的代码示例

按照答案和其他建议我得到了正确的结果，如果你需要它你可以找到完整的代码here .

Answer 1

看起来您需要将此字段添加到您的类DigitPrediction:

public class DigitPrediction
{
    [ColumnName("PredicatedLabel")]
    public uuint ExpectedDigit; // <-- This is the predicted value

    [ColumnName("Score")]
    public float[] Score; // <-- This is the probability that the predicted value is the right classification
}

而且我认为您需要更改将结果写入的行:

    Console.WriteLine($"Predicted digit is: {prediction.ExpectedDigit}");

还有一件事，API 中似乎存在一个错误，其中预期数字将偏离一个，但如果您通过将 +1 加到预测值来移动它，它将是正确的值。我希望他们将来能解决这个问题，它有一个问题:( https://github.com/dotnet/machinelearning/issues/235 )