From 3b16706ba5d52c27fb87b973459a8110ab81ede4 Mon Sep 17 00:00:00 2001
From: Kerry Jiang <kerry-jiang@hotmail.com>
Date: Wed, 24 Jul 2019 19:19:50 -0700
Subject: [PATCH] added a static load method for MnistModelLoader

---
 src/TensorFlowHub/MnistModelLoader.cs | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/src/TensorFlowHub/MnistModelLoader.cs b/src/TensorFlowHub/MnistModelLoader.cs
index 72a4dcbb..d3341f32 100644
--- a/src/TensorFlowHub/MnistModelLoader.cs
+++ b/src/TensorFlowHub/MnistModelLoader.cs
@@ -15,6 +15,16 @@ namespace Tensorflow.Hub
         private const string TEST_IMAGES = "t10k-images-idx3-ubyte.gz";
         private const string TEST_LABELS = "t10k-labels-idx1-ubyte.gz";
 
+        public static async Task<Datasets<MnistDataSet>> LoadAsync(string trainDir, bool oneHot = false)
+        {
+            var loader = new MnistModelLoader();
+            return await loader.LoadAsync(new ModelLoadSetting
+            {
+                TrainDir = trainDir,
+                OneHot = oneHot
+            });
+        }
+
         public async Task<Datasets<MnistDataSet>> LoadAsync(ModelLoadSetting setting)
         {
             if (setting.TrainSize.HasValue && setting.ValidationSize >= setting.TrainSize.Value)