DecisionTreeClassifier model.fit(X, y) 命令返回错误答案

【问题标题】：DecisionTreeClassifier model.fit(X, y) command is returning an errorDecisionTreeClassifier model.fit(X, y) 命令返回错误
【发布时间】：2020-10-30 00:57:29
【问题描述】：

我正在编写一个模型来根据距离预测旅行时间。我得到了

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)

【问题讨论】：

标签： python pandas decision-tree

【解决方案1】：

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

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

【讨论】：