恢复已取消ML.NET训练中的模型

 

 

 

如果您使用ML.NET VS Add-In来训练您的模型，您可能会发现，如果训练突然取消自身，或者您意外取消它，那么这可能会让您崩溃，真的令人沮丧。

碰巧我找到了它生成的临时文件，你仍然可以在你的代码中继续使用，虽然它生成大约10个不同的模型每个算法，所以你需要尝试找到一个最好的（通常是第9或10个）

那么，文件在哪里？

<user>\AppData\Local\Temp\Microsoft.ML.AutoML

唷！

那如何加载它们？下面是一些示例代码：

private static ModelOutput Predict(ModelInput input)
{
    if (predictionEngine == null)
    {
        // Create new MLContext
        MLContext mlContext = new MLContext();

        ITransformer mlModel = mlContext.Model.Load(modelPath, out var modelInputSchema);
        predictionEngine = mlContext.Model.CreatePredictionEngine<ModelInput, ModelOutput>(mlModel);
    }

    // Use model to make prediction on input data
    var result = predictionEngine.Predict(input);
    return result;
}

以下是静态变量的定义：

private static PredictionEngine<ModelInput, ModelOutput> predictionEngine = null;

private static string modelPath = @””;

模型路径是模型的 ZIP 文件。