Question

我建立了这样的管道：

// PurchaseData.TrainingInputColumnNames is string[] containing the input column names
var predictColumn = nameof(PurchaseData.Brand);
var dataProcessPipeline = mlContext.Transforms.Categorical.OneHotEncoding(nameof(PurchaseData.CurrentBrand))
    .Append(mlContext.Transforms.Categorical.OneHotEncoding(nameof(PurchaseData.Gender)))
    .Append(mlContext.Transforms.Concatenate(DefaultColumnNames.Features, PurchaseData.TrainingInputColumnNames))
    .Append(mlContext.Transforms.Conversion.MapValueToKey(outputColumnName: DefaultColumnNames.Label, inputColumnName: predictColumn))
    .Append(mlContext.Transforms.Normalize())
    .Append(mlContext.Transforms.Conversion.MapKeyToValue(("PredictedLabel", DefaultColumnNames.Label)))
    .AppendCacheCheckpoint(mlContext)


IEstimator<ITransformer> trainer = null;
trainer = mlContext.MulticlassClassification.Trainers.StochasticDualCoordinateAscent
    (
    featureColumn: DefaultColumnNames.Features,
    l2Const: 0.0001f, 
    l1Threshold: null,                    
    maxIterations: 200
    );

var trainingPipeline = dataProcessPipeline.Append(trainer);        

var trainedModel = trainingPipeline.Fit(trainingDataView);

还有一个预测班

public class PurchaseDataPrediction
{
    public float[] Score;
    public string PredictedLabel;        
}

当我尝试使用解码标签时

// https://github.com/dotnet/machinelearning/blob/master/test/Microsoft.ML.Tests/Scenarios/Api/Estimators/PredictAndMetadata.cs
VBuffer<ReadOnlyMemory<char>> keys = default;
predictionEngine.OutputSchema[nameof(PurchaseDataPrediction.PredictedLabel)].GetKeyValues(ref keys);

我得到了例外：

“无法将类型为“ Key”的IDataView列“ PredictedLabel”绑定到类型为“ System.String”的字段或属性“ PredictedLabel”。

我在做什么错了？

Answer 1

该PredictAndMetadata的编写思路是您在管道中拥有什么Multiclass训练器，它将产生具有“ Label”列类型的“ PredictedLabel”列。我认为您的管道中没有培训师，我认为它根本不存在。

您这样做：

.Append(mlContext.Transforms.Conversion.MapKeyToValue(("PredictedLabel", DefaultColumnNames.Label)))

采用字符串类型的“标签”，然后将其转换为键类型为“ PredictedLabel”的列。（键基本上是用uint备份的枚举）。

public class PurchaseDataPrediction
{
    public float[] Score;
     public string PredictedLabel;
}

但是您的结果类型定义具有PredictedLabel的字符串类型。在DataView中时，您具有Key（uint）。

这正是异常的意思：

Can't bind the IDataView column 'PredictedLabel' of type 'Key' to field or property 'PredictedLabel' of type 'System.String'.

目前我不确定您要使用此代码实现什么，如果您可以描述要解决的任务，我将为您提供帮助。

Answer 2

这是一个如何将预测标签（作为字符串）取回的示例

        // Create Estimator
        var pipe = mlContext.Transforms.Concatenate("Features", "SepalLength", "SepalWidth", "PetalLength", "PetalWidth")
            .Append(mlContext.Transforms.Normalize("Features"))
            .Append(mlContext.Transforms.Conversion.MapValueToKey("Label", "IrisPlantType"), TransformerScope.TrainTest)
            .AppendCacheCheckpoint(mlContext)
            .Append(mlContext.MulticlassClassification.Trainers.StochasticDualCoordinateAscent())
            .Append(mlContext.Transforms.Conversion.MapKeyToValue(("PredictPlant", "PredictedLabel")));

        // Train the pipeline
        var trainedModel = pipe.Fit(trainData);

        // Make predictions
        var predictFunction = trainedModel.CreatePredictionEngine<IrisDataWithStringLabel, IrisPredictionWithStringLabel>(mlContext);
        IrisPredictionWithStringLabel prediction = predictFunction.Predict(new IrisDataWithStringLabel()
        {
            SepalLength = 5.1f,
            SepalWidth = 3.3f,
            PetalLength = 1.6f,
            PetalWidth = 0.2f,
        });

        // Outputs string : "Iris-setosa" as the prediction
        Console.WriteLine(prediction.PredictPlant);

请注意在管道中指定培训师的位置。另外，在 MapKeyToValue

中指定的位置和参数

正在使用的预测类与以上示例中的类似：

    private class IrisPredictionWithStringLabel
    {
        [ColumnName("Score")]
        public float[] PredictedScores { get; set; }

        public string PredictPlant { get; set; }
    }

希望有帮助！

Answer 3

我认为您的预测班有string PredictedLabel，而我相信GetKeyValues希望有一个关键专栏。

有关键到值转换和值到键转换的更多信息，请参见以下示例： https://github.com/dotnet/machinelearning/blob/master/docs/samples/Microsoft.ML.Samples/Dynamic/KeyToValueValueToKey.cs

异常映射键类型到预测标签

3 个答案: