diff options
| author | Serghei Cebotari <serghei@cebotari.ru> | 2024-01-12 00:23:58 +0300 |
|---|---|---|
| committer | Serghei Cebotari <serghei@cebotari.ru> | 2024-01-12 00:23:58 +0300 |
| commit | fb94d14854b0663f810128d3341911f82441653c (patch) | |
| tree | 018a13e94699a99214921e2778019e0e261e4015 /RhSolutions.ML.Lib | |
| parent | 3fc8fc9db6441f32175e06138881d289e560ac6a (diff) | |
Fix evaluate method
Diffstat (limited to 'RhSolutions.ML.Lib')
| -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() |