def experiment_5():
config = get_experiment_5_config()
logger = Logger(config)
logger.start()
loader = ImageDataLoader(config)
input_shape = config.input_shape
logging.info(f"input_shape:{input_shape}")
number_of_classes = len(loader.labbel_mapper.classes_name)
logging.info(f"number_of_classes:{number_of_classes}")
trainer = Trainer(config=config, loader=loader)
trainer.start(model=get_cnn(hidden_activation=config.hidden_activation,
output_activation=config.output_activation,
input_shape=input_shape, number_of_classes=number_of_classes),
run_id="cnn_model_greyscale")
logger.end()
def main():
config = get_experiment_1_config()
logger = Logger(config)
logger.start()
loader = ImageDataLoader(config)
input_shape = config.input_shape
logging.info(f"input_shape:{input_shape}")
number_of_classes = len(loader.labbel_mapper.labels)
logging.info(f"number_of_classes:{number_of_classes}")
trainer = Trainer(config=config, loader=loader, logger=logger)
trainer.start(model=get_cnn(hidden_activation=config.hidden_activation,
output_activation=config.output_activation,
input_shape=input_shape,
number_of_classes=number_of_classes))
logger.end()
