def main(): training_data, validation_data, test_data = load_data_wrapper() network = Network([784, 15, 10]) network.SGD(training_data=training_data, epochs=30, mini_batch_size=64, eta=0.5, test_data=test_data)
def run(training_data, epochs, mini_batch_size, eta, test_data=None): # Initialize Neural Network print('Initializing neural network') nn = Network([30, 30, 1]) print('Training network') start = time.time() nn.SGD(training_data, epochs, mini_batch_size, eta, test_data=test_data) print('Runtime: {}s'.format(time.time() - start))
def trainNetwork(): training_data, validation_data, test_data = load_data_wrapper() net = Network([784, 60, 30, 10]) net.SGD(training_data, 50, 10, 3.0, test_data=test_data) return net
from src.mnist_loader import load_data_wrapper from src.network import Network if __name__ == '__main__': train, vaild, test = load_data_wrapper() net = Network([784, 30, 10]) net.SGD(train, 30, 10, 3.0, test_data=test)