sklearn函数：train_test_split（分割训练集和测试集）

参考：
https://zhuanlan.zhihu.com/p/248634166