from Classifiers import Classifier
from Classifiers import get_keras_mnist
from Classifiers import build_MyNetBravo


if __name__ == '__main__':
    train_X, train_y, test_X, test_y, data_shape, labels = get_keras_mnist()
    model = build_MyNetBravo(
        data_shape, 
        len(labels), 
        filters=[32, 64, 128, 256], 
        enable_dropout=False, 
        enable_batch_normalization=True
        )
    classifier = Classifier()
    classifier.train_epochs = 1
    classifier.output = 'output/mnist/bravonet-alpha'
    classifier.train(model, train_X, train_y, test_X, test_y, labels)

from Classifiers import Classifier
from Classifiers import get_keras_mnist
from Classifiers import build_MyNetBravo


if __name__ == '__main__':
    train_X, train_y, test_X, test_y, data_shape, labels = get_keras_mnist()
    model = build_MyNetBravo(
        data_shape, 
        len(labels), 
        filters=[32, 64, 128, 256], 
        enable_dropout=False, 
        enable_batch_normalization=True
        )
    classifier = Classifier()
    classifier.train_epochs = 1
    classifier.output = 'output/mnist/bravonet-alpha'
    classifier.train(model, train_X, train_y, test_X, test_y, labels)
