2016-04-25 17 views
1

私は次のpythonプログラムを使用して、基本的な決定木の分類子を実装しています。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) 

それは私が

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

、これを使用してそれを修正しようとしています。しかし、これは同じ警告を示した予測値を出力しますが、警告を与え、

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. 

。助言がありますか?

+0

これらの変更を行うと、同じ警告または別の問題が発生しますか。 –

+0

同じ警告が表示されます。 – user2251234

答えて

3

問題はmodel.predictです。これは動作します: a = model.predict ([[160,0]])