I'm getting stuck with the same problem.The model allways predicts the same class. Did you find an answer for that?