public TensorFlowEstimator(IHostEnvironment env, string model, string[] inputs, string[] outputs) : this(env, new TensorFlowTransform(env, TensorFlowUtils.GetSession(env, model), inputs, outputs, TensorFlowUtils.IsSavedModel(env, model) ? model : null, false)) { }
/// <summary> /// Convenience constructor for public facing API. /// </summary> /// <param name="env">Host Environment.</param> /// <param name="input">Input <see cref="IDataView"/>. This is the output from previous transform or loader.</param> /// <param name="model">Path to the TensorFlow model. </param> /// <param name="names">Name of the output column(s). Keep it same as in the Tensorflow model.</param> /// <param name="source">Name of the input column(s). Keep it same as in the Tensorflow model.</param> public static IDataTransform Create(IHostEnvironment env, IDataView input, string model, string[] names, string[] source) { return(new TensorFlowTransform(env, TensorFlowUtils.GetSession(env, model), source, names, TensorFlowUtils.IsSavedModel(env, model) ? model : null, false).MakeDataTransform(input)); }
// Factory method for SignatureDataTransform. private static IDataTransform Create(IHostEnvironment env, Arguments args, IDataView input) { Contracts.CheckValue(env, nameof(env)); env.CheckValue(args, nameof(args)); env.CheckValue(input, nameof(input)); env.CheckValue(args.InputColumns, nameof(args.InputColumns)); env.CheckValue(args.OutputColumns, nameof(args.OutputColumns)); return(new TensorFlowTransform(env, TensorFlowUtils.GetSession(env, args.Model), args.InputColumns, args.OutputColumns, TensorFlowUtils.IsSavedModel(env, args.Model) ? args.Model : null, false).MakeDataTransform(input)); }