def save_model(self, path): """ It saves the model on disk as well as the weights assigned to the neurons in the network :param path: path where the model and weight files should be stored """ if not os.path.exists(path): self.logger.info('Model - Path [{}] does not exist. Creating...' ''.format(path)) create_dir(path) self.model.save(os.path.join(path, self.name + '_model.h5')) self.model.save_weights(os.path.join(path, self.name + '_weights.h5'))
def main(args): utils.create_dir(args.outputdir) if args.parser_mode == 'dfw_services' or args.parser_mode == 'dfw_appliedto': utils.validate_file(args.rules, args.debug) if args.parser_mode == 'dfw_appliedto': utils.validate_file(args.addrsets, args.debug) args.prefix = '%s-' % (args.prefix) parsedRules = RulesParser(args.rules, args.debug) if args.parser_mode == 'dfw_services': process_services(parsedRules, args) if args.parser_mode == 'dfw_appliedto': process_applied_to(parsedRules, args) # Display any parsing errors at the end parsedRules.parse_errors(args.outputdir, args.prefix)
def save_figure(self, fig, name, path=None, append_date=True): """ Function that saves the figure passed as param in the path given. :param fig: Figure object that will be saved as image :param name: Name of the image :param path: Target path :param append_date: Boolean that specifies if date should be appended to image name """ if not path: path = self.fig_path if not os.path.exists(path): self.logger.info('Model - Path [{}] does not exist. Creating...' ''.format(path)) create_dir(path) if append_date: name = '{}_{}.png'.format(name, datetime.now().strftime('%Y%m%d%H%M%S')) fig.savefig(os.path.join(path, name)) self.logger.info('Model - Figure was saved in [{}]'.format( os.path.join(path, name)))
def prepare_data_in_folders(self, x, y, path): """ It copies images into their corresponding class folders. :param x: list of images paths :param y: list of classes/labels :param path: base path where class folders will be created """ full_path = os.path.abspath(path) if os.path.exists(full_path): # just in case it exists and contains previous executions data remove_dir(full_path) else: create_dir(full_path) i = 0 for f in x: fname = f.split('/')[-1] cls = get_value(y[i], self.classes) dst = os.path.join(full_path, cls) if not os.path.exists(dst): create_dir(dst) dst = os.path.join(dst, fname) copy_file(f, dst) i += 1