Esempi in Python per CNN.evaluate_model

Esempio n. 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')