TensorFlow.NET/test/TensorFlowNET.UnitTest/Hub/MnistModelLoaderTest.cs

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);
        }
    }
}