many fixes, much lower memory consumption on staged_predict_proba