From 3be0df94c032b945af803d5056fb01df7af417f3 Mon Sep 17 00:00:00 2001 From: Serghei Cebotari Date: Fri, 29 Dec 2023 00:05:17 +0300 Subject: Remove RhSolutions.Console --- RhSolutions.ML.Tests/RhSolutionsTests.cs | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) (limited to 'RhSolutions.ML.Tests') diff --git a/RhSolutions.ML.Tests/RhSolutionsTests.cs b/RhSolutions.ML.Tests/RhSolutionsTests.cs index e1ec8f4..4e91342 100644 --- a/RhSolutions.ML.Tests/RhSolutionsTests.cs +++ b/RhSolutions.ML.Tests/RhSolutionsTests.cs @@ -1,3 +1,5 @@ +using RhSolutions.ML.Lib; + namespace RhSolutions.ML.Tests; public abstract class RhSolutionsTests @@ -6,13 +8,13 @@ public abstract class RhSolutionsTests protected static string _dataPath = Path.Combine(_appPath, "..", "..", "..", "..", "Models", "model.zip"); protected MLContext _mlContext; protected PredictionEngine _predEngine; - - [SetUp] - public void Setup() + + public RhSolutionsTests() { + RhSolutionsMLBuilder.RebuildModel(); _mlContext = new MLContext(seed: 0); ITransformer loadedNodel = _mlContext.Model.Load(_dataPath, out var _); - _predEngine = _mlContext.Model.CreatePredictionEngine(loadedNodel); + _predEngine = _mlContext.Model.CreatePredictionEngine(loadedNodel); } public void Execute(string name, string expectedGroup) @@ -23,5 +25,5 @@ public abstract class RhSolutionsTests }; var prediction = _predEngine.Predict(p); Assert.That(prediction.Type, Is.EqualTo(expectedGroup)); - } + } } \ No newline at end of file -- cgit v1.2.3