Skip to content

Commit eafe78d

Browse files
committed
Minor mods to p2b2.py code
1 parent 2dad3de commit eafe78d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Pilot2/P2B2/p2b2.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -270,7 +270,7 @@ def train_ac(self):
270270
x,y=next(imggen)
271271
subset_sample_weight=np.ones((x.shape[0],1))
272272
sample_weight=np.zeros((x.shape[0],self.look_back))
273-
sample_weight[:,0:1]=subset_sample_weight
273+
sample_weight[:,0:look_forward]=subset_sample_weight
274274
loss_data=self.model.train_on_batch(x,y,sample_weight=sample_weight)
275275
iter_loss.append(loss_data)
276276
file_loss.append(np.array(iter_loss).mean(axis=0))

0 commit comments

Comments
 (0)