2 writes to _labelTensor
Microsoft.ML.Vision (2)
ImageClassificationTrainer.cs (2)
1257
_labelTensor
= tf.placeholder(tf.int64, new TensorShape(batch_size), name: labelColumn);
1320
(_trainStep, _crossEntropy,
_labelTensor
, _softMaxTensor) =
11 references to _labelTensor
Microsoft.ML.Vision (11)
ImageClassificationTrainer.cs (11)
628
(_evaluationStep, _) = AddEvaluationStep(_softMaxTensor,
_labelTensor
);
733
if (_session.graph.OperationByName(
_labelTensor
.name.Split(':')[0]) == null)
735
throw Host.ExceptParam(nameof(
_labelTensor
.name), $"'{
_labelTensor
.name}' does not" +
947
runnerInputTensorNames.Add(
_labelTensor
.name);
959
validationEvalRunner = new Runner(_session, new[] { _bottleneckInput.name,
_labelTensor
.name },
979
var labelTensorShape =
_labelTensor
.TensorShape.dims.Select(x => (long)x).ToArray();
1177
return (evalSess,
_labelTensor
, evaluationStep, prediction);
1294
return (null, null,
_labelTensor
, _softMaxTensor);
1300
labels:
_labelTensor
, logits: logits);
1313
return (_trainStep, crossEntropyMean,
_labelTensor
, _softMaxTensor);