Neural network always predicts the same class Neural network always predicts the same class python-3.x python-3.x