diff --git a/nilmtk_contrib/disaggregate/WindowGRU.py b/nilmtk_contrib/disaggregate/WindowGRU.py index 2bf9950..184621e 100644 --- a/nilmtk_contrib/disaggregate/WindowGRU.py +++ b/nilmtk_contrib/disaggregate/WindowGRU.py @@ -46,7 +46,7 @@ def partial_fit(self, train_main, train_appliances, do_preprocessing=True, curre mains = train_main.reshape((-1,self.sequence_length,1)) app_reading = app_df.reshape((-1,1)) filepath = self.file_prefix + "-{}-epoch{}.h5".format( - "_".join(appliance_name.split()), + "_".join(app_name.split()), current_epoch, ) checkpoint = ModelCheckpoint(filepath,monitor='val_loss',verbose=1,save_best_only=True,mode='min')