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
Что делаю не так?
Cat-features указывается индекс категориальных фич, а не их название. То есть, например, нужно писать cat_features =[0, 5, 11]
Обсуждают сегодня