Avoid NaN in feature normalization. (#2186)
This commit is contained in:
@@ -49,6 +49,11 @@ static void ComputeMeanAndInvStd(const float *p, int32_t num_rows,
|
|||||||
(*mean)[i] = t;
|
(*mean)[i] = t;
|
||||||
|
|
||||||
float stddev = std::sqrt(sum_sq[i] / num_rows - t * t);
|
float stddev = std::sqrt(sum_sq[i] / num_rows - t * t);
|
||||||
|
|
||||||
|
if (stddev != stddev) {
|
||||||
|
stddev = 0;
|
||||||
|
}
|
||||||
|
|
||||||
(*inv_stddev)[i] = 1.0f / (stddev + 1e-5f);
|
(*inv_stddev)[i] = 1.0f / (stddev + 1e-5f);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user