Scikit學習不字符串值在KNN

Question

我使用Scikit學會做工作

K最近Neigbour分類：

from sklearn.neighbors import KNeighborsClassifier 

model=KNeighborsClassifier() 
model.fit(train_input,train_labels) 

如果打印我的數據：

print("train_input:") 
print(train_input.iloc[0]) 
print("\n") 
print("train_labels:") 
print(train_labels.iloc[0]) 

我得到這樣的：

train_input: 
PassengerId       1 
Pclass        3 
Name   Braund, Mr. Owen Harris 
Sex        male 
Age         22 
SibSp        1 
Parch        0 
Ticket      A/5 21171 
Fare        7.25 
Cabin        NaN 
Embarked        S 
Name: 0, dtype: object 


train_labels: 
0 

代碼失敗，此錯誤：

ValueError        Traceback (most recent call last) 
<ipython-input-21-1f18eec1e602> in <module>() 
    63 
    64 model=KNeighborsClassifier() 
---> 65 model.fit(train_input,train_labels) 
ValueError: could not convert string to float: 'Q' 

那麼，KNN算法不適用於String值嗎？

如何修改我的數據，使其符合Scikit-Learn中的KNN實現？

Answer 1

對於標稱String功能，請考慮一個熱門編碼：http://scikit-learn.org/stable/modules/generated/sklearn.preprocessing.OneHotEncoder.html。

對於有序String特點，考慮標籤編碼（根據您的功能的理解明智的順序）：http://scikit-learn.org/stable/modules/generated/sklearn.preprocessing.LabelEncoder.html。