2015-04-07 31 views
0

我已經使用python的DecisionTreeRegressor()根據連續的預測器對數據進行細分,並且效果很好。 在當前的項目中,我被要求使用分類數據作爲預測。預測器 - 行業領域,響應 - 收入。 使用DecisionTreeRegressor()扔錯誤如何使用DecisionTreeRegressor()進行分類分割?

「無法更改字符串浮動:工業領域」。

你能否建議是否有什麼辦法可以解決這個問題?

+0

您可以將預測變量和響應編碼爲整數。許多方法可以做到,看看例如到[此鏈接](http://nbviewer.ipython.org/github/rasbt/pattern_classification/blob/master/preprocessing/feature_encoding.ipynb)。希望能幫助到你 – lrnzcig

回答

0

對於這種決策樹,您需要使用DecisionTreeClassifier()。看來DecisionTreeRegressor只能用於預測數據。 DecisionTreeClassifier()僅適用於類預測器數據。我真的想要一個能夠做到這一點的人,但這似乎不可能。