train neural network on multiple feature dimensions