avoid empty dataset (#6756)

dotnet · Jul 6, 2023 · 36f87d1 · 36f87d1
1 parent 26c2446
commit 36f87d1
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/src/Microsoft.ML.AutoML/AutoMLExperiment/IDatasetManager.cs b/src/Microsoft.ML.AutoML/AutoMLExperiment/IDatasetManager.cs
@@ -78,7 +78,15 @@ public IDataView LoadTrainDataset(MLContext context, TrialSettings? settings)
                 var subSampleRatio = parameter.ContainsKey(_subSamplingKey) ? parameter[_subSamplingKey].AsType<double>() : 1;
                 if (subSampleRatio < 1.0)
                 {
-                    var subSampledTrainDataset = context.Data.TakeRows(_trainDataset, (long)(subSampleRatio * _rowCount));
+                    var count = (long)(subSampleRatio * _rowCount);
+                    if (count <= 10)
+                    {
+                        // fix issue https://github.com/dotnet/machinelearning-modelbuilder/issues/2734
+                        // take at least 10 rows to avoid empty dataset
+                        count = 10;
+                    }
+
+                    var subSampledTrainDataset = context.Data.TakeRows(_trainDataset, count);
                     return subSampledTrainDataset;
                 }
             }