diff options
| -rw-r--r-- | RhSolutions.ML.Lib/RhSolutionsMLBuilder.cs | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/RhSolutions.ML.Lib/RhSolutionsMLBuilder.cs b/RhSolutions.ML.Lib/RhSolutionsMLBuilder.cs index b1a7a66..ac8cb9d 100644 --- a/RhSolutions.ML.Lib/RhSolutionsMLBuilder.cs +++ b/RhSolutions.ML.Lib/RhSolutionsMLBuilder.cs @@ -9,7 +9,6 @@ public class RhSolutionsMLBuilder private static string _testDataPath = Path.Combine(_appPath, "..", "..", "..", "..", "TestData", "*"); private static MLContext _mlContext = new MLContext(seed: 0); - private static ITransformer? _trainedModel = null; public static void RebuildModel() { @@ -20,10 +19,13 @@ public class RhSolutionsMLBuilder SaveModelAsFile(_mlContext, _trainDataView.Schema, _trainedModel); } - public static MulticlassClassificationMetrics? EvaluateModel() + public static MulticlassClassificationMetrics EvaluateModel() { var testDataView = _mlContext.Data.LoadFromTextFile<Product>(_testDataPath, hasHeader: false, separatorChar: ';'); - return _mlContext.MulticlassClassification.Evaluate(_trainedModel?.Transform(testDataView)); + MLContext mlContext = new(seed: 0); + string modelPath = Path.Combine(_appPath, "..", "..", "..", "..", "Models", "model.zip"); + var trainedModel = mlContext.Model.Load(modelPath, out _); + return _mlContext.MulticlassClassification.Evaluate(trainedModel.Transform(testDataView)); } private static IEstimator<ITransformer> ProcessData() |