Python CNN.compile_model Examples

Example #1

def main():
    train_data_preprocessor = DataPreprocessor(dataset='train')
    test_data_preprocessor = DataPreprocessor(dataset='test')

    train_data_preprocessor.preprocess()
    test_data_preprocessor.preprocess()

    cnn = CNN()
    cnn.init_model()
    cnn.compile_model()
    cnn.train_model(train_data_preprocessor)
    cnn.evaluate_model(test_data_preprocessor)
    print('done')

Example #2

 def train_top_layers(cls):
     top_layers = CNN.get_top_layers(input_shape=CNN.input_shape())
     top_layers_compiled = CNN.compile_model(top_layers)
     cls.fit_model(top_layers_compiled, save_full=False)