Можем ли мы откалибровать CatboostClassifier, разработанный на числовых и категориальных данных? - PullRequest
0 голосов
/ 20 марта 2020

Я пытаюсь использовать CalibratedClassifierCV в моем классификаторе Catboost, который здесь определен:

X=feature_matrix
y=target_variable

cate_indexes=np.where(X.dtypes =='object')[0]

#definition of classifier: passing indexes of categorical variables
booster_clf = CatBoostClassifier(cat_features=cate_indexes)


model_12_isotonic = CalibratedClassifierCV(booster_clf, cv=5, method='isotonic')
model_12_isotonic.fit(X, y)

Я разрабатываю модель в базе данных, включающую как числовые, так и категориальные переменные, поэтому я объявляю в определении классификатора cat_features input.

Тем не менее, я продолжаю иметь эту ошибку при вызове функции подгонки:

ValueError: could not convert string to float: 'USA'

Это Catboost с категориальными индексами, поддерживаемыми в CalibratedClassifierCV?

Большое спасибо:)

...