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)

Answer 1

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

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