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