basic neural network, trains until a specified number of epochs have
elapsed