|
|
|
@@ -0,0 +1,28 @@ |
|
|
|
using Microsoft.VisualStudio.TestTools.UnitTesting; |
|
|
|
using System; |
|
|
|
using System.Collections.Generic; |
|
|
|
using System.Text; |
|
|
|
using System.Threading.Tasks; |
|
|
|
using Tensorflow; |
|
|
|
using Tensorflow.Hub; |
|
|
|
|
|
|
|
namespace TensorFlowNET.UnitTest.Hub |
|
|
|
{ |
|
|
|
[TestClass] |
|
|
|
public class MnistModelLoaderTest |
|
|
|
{ |
|
|
|
[TestMethod] |
|
|
|
public async Task TestLoad() |
|
|
|
{ |
|
|
|
var loader = new MnistModelLoader(); |
|
|
|
var result = await loader.LoadAsync(new ModelLoadSetting |
|
|
|
{ |
|
|
|
TrainDir = "mnist", |
|
|
|
OneHot = true, |
|
|
|
ValidationSize = 5000, |
|
|
|
}); |
|
|
|
|
|
|
|
Assert.IsNotNull(result); |
|
|
|
} |
|
|
|
} |
|
|
|
} |