summaryrefslogtreecommitdiff
path: root/RhSolutions.ML.Tests/RhSolutionsTests.cs
diff options
context:
space:
mode:
Diffstat (limited to 'RhSolutions.ML.Tests/RhSolutionsTests.cs')
-rw-r--r--RhSolutions.ML.Tests/RhSolutionsTests.cs29
1 files changed, 0 insertions, 29 deletions
diff --git a/RhSolutions.ML.Tests/RhSolutionsTests.cs b/RhSolutions.ML.Tests/RhSolutionsTests.cs
deleted file mode 100644
index 4e91342..0000000
--- a/RhSolutions.ML.Tests/RhSolutionsTests.cs
+++ /dev/null
@@ -1,29 +0,0 @@
-using RhSolutions.ML.Lib;
-
-namespace RhSolutions.ML.Tests;
-
-public abstract class RhSolutionsTests
-{
- protected static string _appPath = Path.GetDirectoryName(Environment.GetCommandLineArgs()[0]) ?? ".";
- protected static string _dataPath = Path.Combine(_appPath, "..", "..", "..", "..", "Models", "model.zip");
- protected MLContext _mlContext;
- protected PredictionEngine<Product, TypePrediction> _predEngine;
-
- public RhSolutionsTests()
- {
- RhSolutionsMLBuilder.RebuildModel();
- _mlContext = new MLContext(seed: 0);
- ITransformer loadedNodel = _mlContext.Model.Load(_dataPath, out var _);
- _predEngine = _mlContext.Model.CreatePredictionEngine<Product, TypePrediction>(loadedNodel);
- }
-
- public void Execute(string name, string expectedGroup)
- {
- Product p = new()
- {
- Name = name
- };
- var prediction = _predEngine.Predict(p);
- Assert.That(prediction.Type, Is.EqualTo(expectedGroup));
- }
-} \ No newline at end of file