Qi Dai

Results 2 comments of Qi Dai

I solve this problem by removing the argument _stratify_ in function `train_test_split ` at the line 306.