Python Machine Learning Warning

Question

Я использую следующую программу python для реализации базового классификатора дерева решений.

from sklearn import tree 
from sklearn.tree import DecisionTreeClassifier 
import numpy as np 

features = [[140,1],[130,1],[150,0],[170,0]] 
labels = [0,0,1,1] 

clf = DecisionTreeClassifier() 
model = clf.fit(features, labels) 
a = model.predict ([160,0]) 
print (a) 

Она выводит прогнозируемое значение, но дает предупреждение,

DeprecationWarning: Passing 1d arrays as data is deprecated in 0.17 and 
willraise ValueError in 0.19. Reshape your data either using X.reshape(-1,  
1) if your data has a single feature or X.reshape(1, -1) if it contains a 
single sample. 

Я попытался исправить это с помощью этого

features = np.array(features).reshape(-1, 2) 
labels = np.array(labels).reshape(-1, 1) 

Но это показало такое же предупреждение. Какие-либо предложения?

Answer 1

Проблема с model.predict. Это работает: a = model.predict ([[160,0]])