staged_predict predicts classes, not probabilites
Merge pull request #2964 from abhishekkrthakur/patch-1

staged_predict predicts classes, not probabilites