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)
