python - (Python) 如何拟合 ValueError: Expected 2D array, got 1D array:

Question

from sklearn.model_selection import train_test_split
target= data.satisfaction
features = data.evaluation
target_train,target_test,features_train,features_test=train_test_split(
target,features,test_size=0.25)
from sklearn.tree import DecisionTreeClassifier
model = DecisionTreeClassifier(random_state=42)
model.fit(features_train,target_train)
model.score(features_train,target_train)*100
model.score(features_test,target_test)*100

问题是 ：

ValueError：预期 2D 数组，得到 1D 数组：array=[0.57 0.86 0.62 ... 0.99 0.96 0.8]。如果您的数据具有单个特征，则使用 array.reshape(-1, 1) 重塑您的数据，如果它包含单个样本，则使用 array.reshape(1, -1) 。

标签： python