diff options
| author | Serghei Cebotari <serghei@cebotari.ru> | 2024-01-11 17:48:01 +0300 |
|---|---|---|
| committer | Serghei Cebotari <serghei@cebotari.ru> | 2024-01-11 17:48:01 +0300 |
| commit | 9c87e6d4da7a4dd08220cd0ec436a3651bc571af (patch) | |
| tree | 8ed6f479f1ddfc38ec5236ea127d6ffca0c23898 /RhSolutions.ML.Tests | |
| parent | 8e17c5b2da4f90be86a0c619d9f8fdbe772851fc (diff) | |
Add ModelBuilder
Diffstat (limited to 'RhSolutions.ML.Tests')
| -rw-r--r-- | RhSolutions.ML.Tests/ModelBuilder.cs | 10 | ||||
| -rw-r--r-- | RhSolutions.ML.Tests/TestBase.cs | 5 |
2 files changed, 11 insertions, 4 deletions
diff --git a/RhSolutions.ML.Tests/ModelBuilder.cs b/RhSolutions.ML.Tests/ModelBuilder.cs new file mode 100644 index 0000000..2fd6115 --- /dev/null +++ b/RhSolutions.ML.Tests/ModelBuilder.cs @@ -0,0 +1,10 @@ +using RhSolutions.ML.Lib; + +namespace RhSolutions.ML.Tests; + +[SetUpFixture] +public class ModelBuilder +{ + [OneTimeSetUp] + public void Build() => RhSolutionsMLBuilder.RebuildModel(); +}
\ No newline at end of file diff --git a/RhSolutions.ML.Tests/TestBase.cs b/RhSolutions.ML.Tests/TestBase.cs index 679d9d3..6295c3f 100644 --- a/RhSolutions.ML.Tests/TestBase.cs +++ b/RhSolutions.ML.Tests/TestBase.cs @@ -1,5 +1,3 @@ -using RhSolutions.ML.Lib; - namespace RhSolutions.ML.Tests; public abstract class TestBase @@ -11,7 +9,6 @@ public abstract class TestBase public TestBase() { - RhSolutionsMLBuilder.RebuildModel(); _mlContext = new MLContext(seed: 0); ITransformer loadedNodel = _mlContext.Model.Load(_dataPath, out var _); _predEngine = _mlContext.Model.CreatePredictionEngine<Product, TypePrediction>(loadedNodel); @@ -36,4 +33,4 @@ public abstract class TestBase var prediction = _predEngine.Predict(actual); Assert.That(prediction.Type, Is.EqualTo(expected.Type)); } -}
\ No newline at end of file +} |