TensorFlow.NET/src/TensorFlowNET.Hub/ModelLoadSetting.cs at master · Oceania2018/TensorFlow.NET · GitHub

20 lines (19 loc) · 596 Bytes

using System;

using System.Collections.Generic;

using System.Text;

using NumSharp;

namespace Tensorflow.Hub

{

public class ModelLoadSetting

{

public string TrainDir { get; set; }

public bool OneHot { get; set; }

public Type DataType { get; set; } = typeof(float);

public bool ReShape { get; set; }

public int ValidationSize { get; set; } = 5000;

public int? TrainSize { get; set; }

public int? TestSize { get; set; }

public string SourceUrl { get; set; }

public bool ShowProgressInConsole { get; set; }

}

Expand file tree