diff --git a/data/gan.py b/data/gan.py index 5559e4d..5fc7032 100644 --- a/data/gan.py +++ b/data/gan.py @@ -536,7 +536,7 @@ class Predict(GNet): self.values = args['values'] self.ROW_COUNT = args['row_count'] self.oROW_COUNT = self.ROW_COUNT - self.MISSING_VALUES = np.nan + self.MISSING_VALUES = np.nan_to_num(np.nan) if 'no_value' in args and args['no_value'] not in ['na','','NA'] : self.MISSING_VALUES = args['no_value']