我用的tensorflow是2.3的版本,然后numpy、scipy啥的版本如果不匹配就容易影响运行。
所以配置如下:
python 3.7+tensorflow 2.3+numpy 1.18.5+scipy 1.6.0+scikit-learn 0.23.2
报错代码如下:
import sklearn
from sklearn.model_selection import train_test_split
X_train, X_validation, Y_train, Y_validation = train_test_split(train_feature_df,train_data.label,test_size = 0.2)