Browse Source

added a static load method for MnistModelLoader

tags/v0.12
Kerry Jiang 6 years ago
parent
commit
a649bd5f8a
1 changed files with 10 additions and 0 deletions
  1. +10
    -0
      src/TensorFlowHub/MnistModelLoader.cs

+ 10
- 0
src/TensorFlowHub/MnistModelLoader.cs View File

@@ -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)


Loading…
Cancel
Save