Python: визуализация дерева решений: IndexError: список индексов вне диапазона - PullRequest
Новая
       9

Python: визуализация дерева решений: IndexError: список индексов вне диапазона

0 голосов
/ 25 мая 2018

Я следовал этому уроку (используя мои собственные данные)

Я до сих пор пытаюсь визуализировать данные в виде графика, но независимо от того, какая метка измой фрейм данных, который я ввожу, говорит, что он выходит за пределы диапазона. 

clusterDF=pd.DataFrame(data=clusterdata[:,:],index=list(range(len(clusterdata))),\
    columns=['viewed','carted','knownpurchases','totlength','avgtime','stdtime','vartime','KMP','Leven','prodnum','Class'])

X = clusterDF.drop('Class', axis=1)  
y = clusterDF['Class']
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.20)
regressor = DecisionTreeRegressor(max_depth=2)  
regressor.fit(X_train, y_train)

y_pred = regressor.predict(X_test)

df=pd.DataFrame({'Actual':y_test, 'Predicted':y_pred})

#Problematic line
export_graphviz(regressor, out_file='foo.dot', feature_names=['carted'])

Полная ошибка..

1 Ответ

0 голосов
/ 25 мая 2018

Вам необходимо указать название всех ваших функций: 

feature_names = ['viewed','carted','knownpurchases','totlength','avgtime','stdtime','vartime','KMP','Leven','prodnum']

В руководстве используется только 1 функция (пояснительная переменная).

...