Команда DecisionTreeClassifier model.fit (X, y) возвращает ошибку - PullRequest
0 голосов
/ 09 июля 2020

Я кодирую модель для прогнозирования времени в пути на основе расстояния. Я получаю

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)

1 Ответ

1 голос
/ 09 июля 2020

Это потому, что ваши цели являются значениями с плавающей запятой, а не целыми. Возможно, вы пытаетесь классифицировать непрерывные цели.

Возможно, попробуйте DecisionTreeRegressor() или преобразовать ваши цели в целые числа, если ваши метки по ошибке были преобразованы в плавающие. Или дважды проверьте, что вы используете правильные столбцы.

...