在训练数据中:x的数据类型为“ int64”,y的数据类型为“类别” 我无法在此算法中拟合训练数据集。
我在“ model.fit(x_train,y_train)”行出现错误
错误是“类型为'CategoricalDtype'的对象没有len()”
x=df.drop("income",axis=1)
y=df["income"]
from sklearn.model_selection import train_test_split
x_train,x_test,y_train,y_test=train_test_split(x,y,train_size=0.3,random_state=100)
from sklearn.tree import DecisionTreeClassifier
model=DecisionTreeClassifier(max_depth=4)
model.fit(x_train,y_train)
答案 0 :(得分:0)
尝试在.tolist()
之后添加y_train and y_test