27 lines
842 B
C#
27 lines
842 B
C#
|
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;
|
||
|
|
|
||
|
|
[SetUp]
|
||
|
|
public void Setup()
|
||
|
|
{
|
||
|
|
_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));
|
||
|
|
}
|
||
|
|
}
|