如何对 ML.NET 预测进行单元测试？

Question

我有一个 ML.NET 项目部署为这样的 Web api - https://docs.microsoft.com/en-us/dotnet/machine-learning/how-to-guides/serve-model-web-api-ml-net

如何在 xUnit 或任何单元测试框架中模拟 PredictionEnginePool、MLContext 或 PredictionEngine？

Answer 1

您看过 ML.NET 的测试吗？ MLContext 显示在这些示例中。 https://github.com/dotnet/machinelearning/tree/master/test

PredictionEnginePool（我不相信显示），但它不是无法测试的特殊构造。如果需要，我会测试 FileLoader 并能够从远程位置进行测试。