Getting input, training neural network,