update

gdy-scu · Jul 22, 2024 · d928417 · d928417
1 parent 864c35e
commit d928417
Showing 1 changed file with 14 additions and 14 deletions.
diff --git a/Stage3/dataloader.py b/Stage3/dataloader.py
@@ -157,7 +157,7 @@ def prepare_minibatch(self, seqs):
                 raw_seq_spdy.append(int(spdy))
                 raw_seq_spdz.append(int(spdz))
 
-            for step in range(0, self.configs.inp_seq_len + self.configs.horizon):
+            for step in range(1, self.configs.inp_seq_len + self.configs.horizon):
                 t_lon.append(self.convert_tar((raw_seq_lon[step], raw_seq_lon[step - 1]), 'lon'))
                 t_lat.append(self.convert_tar((raw_seq_lat[step], raw_seq_lat[step - 1]), 'lat'))
                 t_alt.append(self.convert_tar((raw_seq_alt[step], raw_seq_alt[step - 1]), 'alt'))
@@ -172,19 +172,19 @@ def prepare_minibatch(self, seqs):
             batch_spdy.append(seq_spdy[:self.configs.inp_seq_len])
             batch_spdz.append(seq_spdz[:self.configs.inp_seq_len])
 
-            batch_t_lon.append(t_lon[self.configs.inp_seq_len:])
-            batch_t_lat.append(t_lat[self.configs.inp_seq_len:])
-            batch_t_alt.append(t_alt[self.configs.inp_seq_len:])
-            batch_t_spdx.append(t_spdx[self.configs.inp_seq_len:])
-            batch_t_spdy.append(t_spdy[self.configs.inp_seq_len:])
-            batch_t_spdz.append(t_spdz[self.configs.inp_seq_len:])
-
-            batch_dec_lon.append(t_lon[:self.configs.inp_seq_len])
-            batch_dec_lat.append(t_lat[:self.configs.inp_seq_len])
-            batch_dec_alt.append(t_alt[:self.configs.inp_seq_len])
-            batch_dec_spdx.append(t_spdx[:self.configs.inp_seq_len])
-            batch_dec_spdy.append(t_spdy[:self.configs.inp_seq_len])
-            batch_dec_spdz.append(t_spdz[:self.configs.inp_seq_len])
+            batch_t_lon.append(t_lon[self.configs.inp_seq_len - 1:])
+            batch_t_lat.append(t_lat[self.configs.inp_seq_len - 1:])
+            batch_t_alt.append(t_alt[self.configs.inp_seq_len - 1:])
+            batch_t_spdx.append(t_spdx[self.configs.inp_seq_len - 1:])
+            batch_t_spdy.append(t_spdy[self.configs.inp_seq_len - 1:])
+            batch_t_spdz.append(t_spdz[self.configs.inp_seq_len - 1:])
+
+            batch_dec_lon.append(t_lon[:self.configs.inp_seq_len - 1])
+            batch_dec_lat.append(t_lat[:self.configs.inp_seq_len - 1])
+            batch_dec_alt.append(t_alt[:self.configs.inp_seq_len - 1])
+            batch_dec_spdx.append(t_spdx[:self.configs.inp_seq_len - 1])
+            batch_dec_spdy.append(t_spdy[:self.configs.inp_seq_len - 1])
+            batch_dec_spdz.append(t_spdz[:self.configs.inp_seq_len - 1])
 
             raw_batch_lon.append(raw_seq_lon)
             raw_batch_lat.append(raw_seq_lat)