2017-02-25 44 views
0

我想擴大我的編程技能,並認爲我會嘗試機器學習。所以這是爲了學習,而不是任何嚴肅的事情。據說,我從sqlite數據庫檢索一些信息,然後嘗試通過RandomForestClassifier運行它,但出現錯誤。Python RandomForestClassifer - 未知的標籤類型:'連續'錯誤

line 172, in check_classification_targets raise ValueError("Unknown label type: %r" % y_type) ValueError: Unknown label type: 'continuous'"

我的代碼如下:

series= cur.fetchall() 
y = [x[1] for x in series] 
x = [x[2] for x in series] 
y = array(y).astype(float) 
x = array(x).astype(int) 
rf_model = RandomForestClassifier() 
rf_model.fit(x, y) 

我的陣列形狀: y.shape (50) x.shape (50)

什麼我錯過了嗎?搜索SO它似乎需要字符串格式的y變量,但我仍然得到錯誤:

"number of samples=%d" % (len(y), n_samples)) ValueError: Number of labels=50 does not match number of samples=1

回答

0

想通了。

我需要轉換陣列X = X[:, None]