Skip to content

Commit 36f87d1

Browse files
avoid empty dataset (#6756)
1 parent 26c2446 commit 36f87d1

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

src/Microsoft.ML.AutoML/AutoMLExperiment/IDatasetManager.cs

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,15 @@ public IDataView LoadTrainDataset(MLContext context, TrialSettings? settings)
7878
var subSampleRatio = parameter.ContainsKey(_subSamplingKey) ? parameter[_subSamplingKey].AsType<double>() : 1;
7979
if (subSampleRatio < 1.0)
8080
{
81-
var subSampledTrainDataset = context.Data.TakeRows(_trainDataset, (long)(subSampleRatio * _rowCount));
81+
var count = (long)(subSampleRatio * _rowCount);
82+
if (count <= 10)
83+
{
84+
// fix issue https://github.com/dotnet/machinelearning-modelbuilder/issues/2734
85+
// take at least 10 rows to avoid empty dataset
86+
count = 10;
87+
}
88+
89+
var subSampledTrainDataset = context.Data.TakeRows(_trainDataset, count);
8290
return subSampledTrainDataset;
8391
}
8492
}

0 commit comments

Comments
 (0)