train_test_split : stratify can not be recognized?
You need to pass an array containing the class-labels (or whatever the criterion for stratifying is) as an argument to stratify. 

In your case, the answer is probably loan['Loan_Status'].values.