Remove redundant predict_leaf
Take is faster than __getitem__