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")
