Skip to content

Commit cc0b786

Browse files
authored
Do not pass validation set and metrics callback to Image Classification API. (#4430)
* Do not pass validation set to Image Classification API. * remove metric callback since it is set to default in ImageClassification Trainer.
1 parent 7a8f063 commit cc0b786

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

src/Microsoft.ML.AutoML/TrainerExtensions/MultiTrainerExtensions.cs

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -240,9 +240,6 @@ public ITrainerEstimator CreateInstance(MLContext mlContext, IEnumerable<Sweepab
240240
ColumnInformation columnInfo, IDataView validationSet)
241241
{
242242
var options = TrainerExtensionUtil.CreateOptions<Options>(null, columnInfo.LabelColumnName);
243-
options.ValidationSet = validationSet;
244-
var logger = ((IChannelProvider)mlContext).Start(nameof(ImageClassificationExtension));
245-
options.MetricsCallback = (ImageClassificationMetrics metric) => { logger.Trace(metric.ToString()); } ;
246243
return mlContext.MulticlassClassification.Trainers.ImageClassification(options);
247244
}
248245

0 commit comments

Comments
 (0)