Cataboost модель обучения - PullRequest
Новая
       58

Cataboost модель обучения

0 голосов
/ 28 декабря 2018

Я использую catboost для обучения моей модели, которая выдает ошибку типа: Код: 

x_train = x_train.astype(np.float32)
x_test = x_test.astype(np.float32)
y_train = y_train.astype(np.float32)
y_test = y_test.astype(np.float32)

train_pool = Pool(x_train, y_train, cat_features=[]) # cat_features=[0,2,5])
test_pool = Pool(x_test)

cb = CatBoostRegressor(iterations=400,
                         learning_rate=0.02,
                         depth=12,
                         eval_metric='RMSE')
cb.fit(train_pool)

Ошибка: 

/opt/conda/lib/python3.6/site-packages/catboost/core.py in _init(self, data, label, cat_features, pairs, weight, group_id, group_weight, subgroup_id, pairs_weight, baseline, feature_names)
    635         if feature_names is not None:
    636             self._check_feature_names(feature_names, features_count)
--> 637         self._init_pool(data, label, cat_features, pairs, weight, group_id, group_weight, subgroup_id, pairs_weight, baseline, feature_names)
    638 
    639 

_catboost.pyx in _catboost._PoolBase._init_pool()

_catboost.pyx in _catboost._PoolBase._init_pool()

TypeError: unhashable type: 'numpy.ndarray'

Я преобразовал все в float32 и всестолбцы числовые.Может кто-нибудь помочь мне разобраться в проблеме здесь?

...