0
0
This commit is contained in:
Serghei Cebotari 2024-01-12 00:01:11 +03:00
parent ac86f7ff28
commit e0931c2e5f

View File

@ -10,8 +10,8 @@ public abstract class TestBase
public TestBase() public TestBase()
{ {
_mlContext = new MLContext(seed: 0); _mlContext = new MLContext(seed: 0);
ITransformer loadedNodel = _mlContext.Model.Load(_dataPath, out var _); ITransformer loadedModel = _mlContext.Model.Load(_dataPath, out var _);
_predEngine = _mlContext.Model.CreatePredictionEngine<Product, TypePrediction>(loadedNodel); _predEngine = _mlContext.Model.CreatePredictionEngine<Product, TypePrediction>(loadedModel);
} }
public void Execute(string name, string expectedGroup) public void Execute(string name, string expectedGroup)