python prediction code for random forest