樸素貝葉斯scikit學習警告「DeprecationWarning：傳遞1d數組」

Question

我知道傳遞1d數組已棄用Naive Bayes分類器，我不知道爲什麼。然而，出於某些原因，一旦我建立了我的模型，我想循環遍歷項目列表，並且對於每個項目我應該調用樸素分類器來預測類。所以，假設我的訓練集是X，我的目標類集是Y：然後我想調用分類器來獲取樣本xx。

X = np.array([[1,0,1,1],[1,1,1,0],[0,0,0,1],[1,0,1,0]) 

Y = np.array([1,1,2,2]) 

from sklearn.naive_bayes import GaussianNB 
clf = GaussianNB() 
clf.fit(X, Y) 
GaussianNB() 

xx=[1,1,0,0] 
clf.predict_proba(xx) 

我該如何解決我的代碼，以便擺脫此警告？

Answer 1

您的測試數據應具有相同的結構，你的訓練數據，修改您的測試xx到：

xx=[[1,1,0,0]] 

將擺脫的警告。