def create_models(dataset_dir): """Initialize the app (available for localhost only) Parameters: dataset_dir (:func:`str`): Path to the training set """ logger.debug("Creating models...") if not local_exec: logger.error("Models can only be generated locally") exit(1) # Modify the configuration for local execution app_config['root'] = os.environ['ROOT'] # Generate inline models and train classifier denoiser = Denoiser(app_config) if not exists(dataset_dir) or not isdir(dataset_dir): logger.error(dataset_dir+" is not a valid directory") exit(2) dataset = [join(dataset_dir, f) for f in listdir(dataset_dir)] denoiser.generate_models(dataset) logger.info("Inline models generated") denoiser.train(dataset) logger.info("Classifier trained")
def create_models(dataset_dir): """Initialize the app (available for localhost only) Parameters: dataset_dir (:func:`str`): Path to the training set """ logger.debug("Creating models...") if not local_exec: logger.error("Models can only be generated locally") exit(1) # Modify the configuration for local execution app_config['root'] = os.environ['ROOT'] # Generate inline models and train classifier denoiser = Denoiser(app_config) if not exists(dataset_dir) or not isdir(dataset_dir): logger.error(dataset_dir + " is not a valid directory") exit(2) dataset = [join(dataset_dir, f) for f in listdir(dataset_dir)] denoiser.generate_models(dataset) logger.info("Inline models generated") denoiser.train(dataset) logger.info("Classifier trained")