Использование PermutationImportance с LGBMClassifier вызывает ValueError: Неизвестный тип метки: «непрерывный» - PullRequest
0 голосов
/ 16 ноября 2018

Я хотел бы получить оценку возможностей eli5, но продолжаю получать ValueError: Unknown label type: 'continuous' Как я могу это исправить?

import eli5
from eli5.sklearn import PermutationImportance
from sklearn.svm import SVC
from sklearn.feature_selection import SelectFromModel
my_model = RandomForestClassifier()

# ... load data
clf = lgb.LGBMClassifier(nthread=4,            boosting_type= 'gbdt', 
            metric= 'auc', n_estimators= 5000    )
perm = PermutationImportance(clf, cv=5)
perm.fit(df[feat], df.CSI)

# perm.feature_importances_ attribute is now available, it can be used
# for feature selection - let's e.g. select features which increase
# accuracy by at least 0.05:
sel = SelectFromModel(perm, threshold=0.05, prefit=True)
X_trans = sel.transform(X)

1 Ответ

0 голосов
/ 16 ноября 2018

Исправлена ​​проблема:

perm.fit(df[feat].values, df.CSI.values)
...