Поиск гиперпарам Catboost - PullRequest
Новая
       90

Поиск гиперпарам Catboost

0 голосов
/ 13 июля 2020

Я хочу использовать гиперпараммы по умолчанию в случайном поиске, как я могу это сделать? (здесь параметр per_float_feature_quantization) 

grid = {'learning_rate': [0.1, 0.16, 0.2],
        'depth': [4, 6, 10],
        'l2_leaf_reg': [1, 3, 5, 7, 9],
        'iterations': [800, 1000, 1500, 2000],
        'bagging_temperature': [1, 2, 3, 4, 5],
        'border_count': [128, 256, 512],
        'grow_policy': ['SymmetricTree', 'Depthwise'],
        'per_float_feature_quantization':[None, '3:border_count=1024']}

model = CatBoostClassifier(loss_function='MultiClass',
                          custom_metric='Accuracy',
                          eval_metric='TotalF1',
                          od_type='Iter',
                          od_wait=40,
                          task_type="GPU", 
                          devices='0:1',
                          random_seed=42,
                          cat_features=cat_features)

randomized_search_result = model.randomized_search(grid,
                                                   X=X,
                                                   y=y
                                                   )

И у меня есть 

CatBoostError: library/cpp/json/writer/json_value.cpp:499: Not a map
...