568_options = options;

553options.TrainSetBottleneckCachedValuesFileName = _options.TrainSetBottleneckCachedValuesFileName; 559options.ValidationSetBottleneckCachedValuesFileName = _options.ValidationSetBottleneckCachedValuesFileName; 569_useLRScheduling = _options.LearningRateScheduler != null; 570_checkpointPath = Path.Combine(_options.WorkspacePath, _options.FinalModelPrefix + 571ModelFileName[_options.Arch]); 572_sizeFile = Path.Combine(_options.WorkspacePath, "TrainingSetSize.txt"); 575var arch = _options.Arch; 600var labelColumn = input.Schema.GetColumnOrNull(_options.LabelColumnName).Value; 609var msg = $"Only one class found in the {_options.LabelColumnName} column. To build a multiclass classification model, the number of classes needs to be 2 or greater"; 613var imageSize = ImagePreprocessingSize[_options.Arch]; 614_session = LoadTensorFlowSessionFromMetaGraph(Host, _options.Arch).Session; 621AddTransferLearningLayer(_options.LabelColumnName, _options.ScoreColumnName, _options.LearningRate, 634bool success = inputSchema.TryFindColumn(_options.LabelColumnName, out _); 642new SchemaShape.Column(_options.ScoreColumnName, SchemaShape.Column.VectorKind.Vector, 644new SchemaShape.Column(_options.PredictedLabelColumnName, SchemaShape.Column.VectorKind.Scalar, 652FeatureColumn.Name, LabelColumn.Name, _options.ScoreColumnName, _options.PredictedLabelColumnName); 659if (!Directory.Exists(_options.WorkspacePath)) 661Directory.CreateDirectory(_options.WorkspacePath); 665CheckTrainingParameters(_options); 666var validationSet = trainContext.ValidationSet?.Data ?? _options.ValidationSet; 668string trainSetBottleneckCachedValuesFilePath = Path.Combine(_options.WorkspacePath, 669_options.TrainSetBottleneckCachedValuesFileName); 671string validationSetBottleneckCachedValuesFilePath = Path.Combine(_options.WorkspacePath, 672_options.ValidationSetBottleneckCachedValuesFileName); 674bool needValidationSet = _options.EarlyStoppingCriteria != null || _options.MetricsCallback != null; 675bool validationSetPresent = _options.ReuseValidationSetBottleneckCachedValues && 681if (generateValidationSet && _options.ValidationSet != null) 683CacheFeaturizedImagesToDisk(validationSet, _options.LabelColumnName, 684_options.FeatureColumnName, imageProcessor, _inputTensorName, _bottleneckTensor.name, 686ImageClassificationMetrics.Dataset.Validation, _options.MetricsCallback); 692if (!_options.ReuseTrainSetBottleneckCachedValues || 696CacheFeaturizedImagesToDisk(trainContext.TrainingSet.Data, _options.LabelColumnName, 697_options.FeatureColumnName, imageProcessor, 699ImageClassificationMetrics.Dataset.Train, _options.MetricsCallback, 700generateValidationSet ? _options.ValidationSetFraction : null); 703(generateValidationSet && _options.ValidationSetFraction.HasValue); 708if (generateValidationSet && _options.ReuseTrainSetBottleneckCachedValues && 709!_options.ReuseValidationSetBottleneckCachedValues) 717validationSetPresent && (_options.EarlyStoppingCriteria != null || _options.MetricsCallback != null) ? 856Path.Combine(_options.WorkspacePath, _options.ValidationSetBottleneckCachedValuesFileName), 911Contracts.Assert(_options.EarlyStoppingCriteria == null || validationNeeded); 922int batchSize = _options.BatchSize; 923int epochs = _options.Epoch; 924float learningRate = _options.LearningRate; 925Action<ImageClassificationMetrics> statisticsCallback = _options.MetricsCallback; 932if (_options.LearningRateScheduler != null) 935if (statisticsCallback != null && _options.TestOnTrainSet) 954trainWriter = tf.summary.FileWriter(Path.Combine(_options.WorkspacePath, "train"), 981BatchSize = _options.BatchSize, 982BatchesPerEpoch = trainingExamples / _options.BatchSize 993_options.LearningRateScheduler, trainState, runner, featureBufferPtr, labelBufferPtr, 996if (_options.TestOnTrainSet && statisticsCallback != null) 1007if (_options.TestOnTrainSet && statisticsCallback != null) 1045if (_options.EarlyStoppingCriteria != null) 1047if (_options.EarlyStoppingCriteria.ShouldStop(metrics.Train)) 1134if (_cleanupWorkspace && Directory.Exists(_options.WorkspacePath)) 1138Directory.Delete(_options.WorkspacePath, true); 1149var evalGraph = LoadMetaGraph(Path.Combine(_resourcePath, ModelFileName[_options.Arch])); 1155var (_, _, groundTruthInput, finalTensor) = AddFinalRetrainOps(classCount, _options.LabelColumnName, 1156_options.ScoreColumnName, bottleneckTensor, false, (_options.LearningRateScheduler == null ? false : true), _options.LearningRate); 1159var imageSize = ImagePreprocessingSize[_options.Arch];