OptCAT(=Optuna+CatBoost)提供了一个scikitlearn兼容的估计器,它使用optnuna调整CatBoost中的超参数。
optcat的Python项目详细描述
OptCAT公司
OptCAT(=Optuna+CatBoost)提供了一个scikit学习兼容估计器,它使用Optuna调整CatBoost中的超参数。在
此存储库受Y-oHr-N/OptGBM的影响很大。在
示例
from optcat.core import CatBoostClassifier from sklearn import datasets params = { "bootstrap_type": "Bayesian", "loss_function": "Logloss", "iterations": 100 } model = CatBoostClassifier(params=params, n_trials=5) data, target = datasets.load_breast_cancer(return_X_y=True) model.fit(X=data, y=target)
安装
^{pr2}$测试
poetry run pytest
- 项目
标签: