diff --git a/2_norm_consts.ipynb b/2_norm_consts.ipynb index 2fb05496a6ee2500e3979335869dcf50ddd5b7d6..a689ae0fff64d45fdfc77502b6e0388f8bc7b0f8 100644 --- a/2_norm_consts.ipynb +++ b/2_norm_consts.ipynb @@ -47,6 +47,7 @@ " y,m,d=f.split('_')\n", " if int(y) in train and int(m) in months:\n", " t_array = np.load(DATADIR+var+'/'+f)\n", + " t_array[np.isnan(t_array)] = 0\n", " tmp += list(t_array.flatten())\n", " mean, std = np.mean(tmp), np.std(tmp)\n", " print(f'Mean {mean}, std {std}')\n", @@ -67,7 +68,8 @@ " for f in tqdm(files):\n", " y,m,d=f.split('_')\n", " if int(y) in train and int(m) in months:\n", - " t_array = np.load(FIREDATADIR+var+'/'+f) \n", + " t_array = np.load(FIREDATADIR+var+'/'+f)\n", + " t_array[np.isnan(t_array)] = 0 \n", " tmp += list(t_array.flatten())\n", " mean, std = np.mean(tmp), np.std(tmp)\n", " print(f'Mean {mean}, std {std}')\n",