ValueError: не может обработать смесь непрерывного и мультикласса - PullRequest
0 голосов
/ 22 декабря 2018

Я хочу оценить модель по данным, которые я использовал здесь в scikit-learn.Я использую функцию DecisionTreeClassifier.score, но при выполнении кода я получу ошибку ValueError:

Не могу обработать смесь непрерывного и мультикласса.

Воткод, который я использую:

from sklearn import datasets
import pandas as pd
import matplotlib.pyplot as plt
import numpy as np
from sklearn.model_selection import train_test_split
from sklearn.tree import DecisionTreeClassifier
nba = pd.read_excel(r"C:\Users\user\Desktop\nba.xlsx")
X = nba.drop('平均得分', axis = 1)
y = nba['平均得分']
X_train, X_test, y_train, y_test = train_test_split(
X, y, test_size = 0.20)
nba_tree = DecisionTreeClassifier()
nba_tree.fit(X_train, y_train.astype('int'))
y_pred = nba_tree.predict(X_test)
nba_tree.score(X_test, y_test)

1 Ответ

0 голосов
/ 22 декабря 2018

Похоже, ваша целевая переменная 平均得分 является непрерывной переменной.Вероятно, вы пытаетесь решить проблему регрессии.Если это так, попробуйте DecisionTreeRegressor вместо DecisionTreeClassifier.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...