diff options
Diffstat (limited to 'RhSolutions.ML.Tests/RhSolutionsTests.cs')
-rw-r--r-- | RhSolutions.ML.Tests/RhSolutionsTests.cs | 29 |
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 |