python - DecisionTreeClassifier model.fit(X, y) 命令返回错误

Question

我正在编写一个模型来根据距离预测旅行时间。我正进入（状态

ValueError：未知标签类型：运行此代码时出现“连续”错误：

import pandas as pd

from sklearn.tree import DecisionTreeClassifier
times = pd.read_csv('SC.csv')
X = times.drop(columns=['Time'])
y = times.drop(columns=['distance'])

model = DecisionTreeClassifier()
model.fit(X, y)

标签： pythonpandasdecision-tree

Answer 1

这是因为您的目标是浮点值，而不是整数。您可能正在尝试对连续目标进行分类。

DecisionTreeRegressor()如果您的标签错误地转换为浮点数，也许可以尝试，或者将您的目标转换为整数。或仔细检查您使用的是正确的列。