Update dataloader.py

fix bugs for mean vector estimation

QFA/dataloader.py

@@ -108,7 +108,7 @@ def __init__(self, config: CN):
         self.data_size = self.flux.shape[0]
 
         s = np.hstack((np.exp(1*self._tau(self.zabs)), np.ones((self.data_size, self.Nr), dtype=float)))
-        self._mu = np.sum(self.flux*s, axis=0)/np.sum(self.flux!=0., axis=0)
+        self._mu = np.sum(self.flux*s*self.mask, axis=0)/np.sum(self.flux!=-999., axis=0)
         self._mu = smooth(self._mu, window_len=config.TRAIN.WINDOW_LENGTH_FOR_MU)
 
     def have_next_batch(self):
@@ -188,4 +188,4 @@ def __getitem__(self, idx):
 
     @property
     def mu(self):
-        return self._mu
+        return self._mu