ENH: Better interfaces in neural network.