AlvLouWeh2017: Replace non-finite features by constant

This commit is contained in:
2021-08-04 13:54:14 -05:00
parent e72f3b553f
commit 4a52911924

View File

@@ -320,8 +320,10 @@ class FeaturesExtractor:
else: else:
f.append(0.0) f.append(0.0)
for v in f: for (i, v) in enumerate(f):
assert isfinite(v), f"non-finite elements detected: {f}" if not isfinite(v):
f[i] = 0.0
features.append(f) features.append(f)
return features return features