This should help you debug/correct your codes 
remember your data for your code assignment cell has been recalled as X and not data_treatment or data_control. Also you are suppose to use predict_proba when you used your previous self recalled functions.