Коллеги, собираю модель cat_features = ['action', 'operation', 'platform', 'status'] cv_dataset = Pool(data=train_pool,

Question

Коллеги, собираю модель cat_features = ['action', 'operation', 'platform', 'status'] cv_dataset = Pool(data=train_pool,

label=train_label.tolist(),
cat_features=cat_features)

params = {"iterations": 100,
"depth": 2,
"loss_function": "Logloss",
"verbose": False}

scores = cv(train_data,
params,
fold_count=5,
plot="True")

catboost.cv пишет:
CatBoostError: Invalid data type={}: data must be list(), np.ndarray(), DataFrame(), Series(), FeaturesData scipy.sparse matrix or filename str().
на train_label.tolist() и cat_features

Что делаю не так?

#database #programming #russian #software

0

21.11.2019

1 ответов

18 просмотров

Blen obema · Accepted Answer

Blen obema

Cat-features указывается индекс категориальных фич, а не их название. То есть, например, нужно писать cat_features =[0, 5, 11]

0

21.11.2019

215 похожих чатов

Коллеги, собираю модель cat_features = ['action', 'operation', 'platform', 'status'] cv_dataset = Pool(data=train_pool,

1 ответов

Похожие вопросы