def plot(self, note=""): DataUtils.message("Ploting Model...", new=True) directory = "plot/" DataUtils.create_dir(directory) file = DataUtils.get_filename("UFF", note)+".png" plot_model(self.model, to_file=directory+file, show_shapes=True, show_layer_names=False)
def save(self, note=""): DataUtils.message("Saving Model...", new=True) directory = "weights/" DataUtils.create_dir(directory) file = DataUtils.get_filename("UFF", note)+".h5" self.model.save(directory+file)