Выбор функций Python

Question

У меня ошибка при выполнении выбора функций в Python.Я новичок в Python.Проблема:

from sklearn.feature_selection import RFE
from sklearn.linear_model import LogisticRegression
array = df.values
X = array[:,0:49]
Y = array[:,48]
# feature extraction
model = LogisticRegression()
rfe = RFE(model, 20)
fit = rfe.fit(X, Y)
print("Num Features: %d") % fit.n_features_
print("Selected Features: %s") % fit.support_
print("Feature Ranking: %s") % fit.ranking_

, когда я запускаю этот код, он показывает ошибку

ValueError: could not convert the string to float: 'High Prone'

Моя целевая переменная в df - «Высокая вероятность» и «Меньшая вероятность».Я сделал факторинг, но он не работает.

Answer 1

Возможно, вам нужно sklearn.preprocessing.LabelEncoder для кодирования целевой переменной в числовые значения.

from sklearn import preprocessing
le = preprocessing.LabelEncoder()
y_encoded = le.fit_transform(y)
fit = rte.fit(X, y_encoded)