首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >打印CatBoost超参数

打印CatBoost超参数
EN

Stack Overflow用户
提问于 2019-03-31 19:45:28
回答 2查看 563关注 0票数 1

如何在训练模型后打印CatBoost超参数?

sklearn中,我们只能打印模型对象,它将显示所有参数,但在catboost中,它只能打印对象的引用:<catboost.core.CatBoostRegressor object at 0x7fd441e5f6d8>

代码语言:javascript
复制
from catboost import CatBoostRegressor
# Initialize data

train_data = [[1, 4, 5, 6],
              [4, 5, 6, 7],
              [30, 40, 50, 60]]

eval_data = [[2, 4, 6, 8],
             [1, 4, 50, 60]]

train_labels = [10, 20, 30]
# Initialize CatBoostRegressor
model = CatBoostRegressor(iterations=2,
                          learning_rate=1,
                          depth=2)
# Fit model
model.fit(train_data, train_labels)
# Get predictions
preds = model.predict(eval_data)
print (model)
EN

回答 2

Stack Overflow用户

回答已采纳

发布于 2019-04-04 19:32:42

print(model.get_params())应该做的是

票数 2
EN

Stack Overflow用户

发布于 2021-01-28 19:36:32

使用print(model.get_all_params())打印所有默认参数。

票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/55440566

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档