如何修复“类型为'CategoricalDtype'的对象没有len()”类型的错误?

时间:2019-02-07 16:02:04

标签: python machine-learning

在训练数据中: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)

1 个答案:

答案 0 :(得分:0)

尝试在.tolist()之后添加y_train and y_test