Skip to content

Commit 6957d2f

Browse files
daholsteDmitry-A
authored andcommitted
fix SMAC bug that causes multiclass sample to infinite loop (dotnet#209)
1 parent ea96f82 commit 6957d2f

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/Microsoft.ML.Auto/Sweepers/SmacSweeper.cs

+4
Original file line numberDiff line numberDiff line change
@@ -386,6 +386,10 @@ private double ComputeEI(double bestVal, double[] forestStatistics)
386386
double empMean = forestStatistics[0];
387387
double empStdDev = forestStatistics[1];
388388
double centered = empMean - bestVal;
389+
if (empStdDev == 0)
390+
{
391+
return centered;
392+
}
389393
double ztrans = centered / empStdDev;
390394
return centered * SweeperProbabilityUtils.StdNormalCdf(ztrans) + empStdDev * SweeperProbabilityUtils.StdNormalPdf(ztrans);
391395
}

0 commit comments

Comments
 (0)