https://github.com/wakame1367/optcat

OptCAT (= Optuna + CatBoost)
https://github.com/wakame1367/optcat

python

Last synced: about 2 months ago
JSON representation

OptCAT (= Optuna + CatBoost)

• Host: GitHub
• URL: https://github.com/wakame1367/optcat
• Owner: wakame1367
• License: mit
• Created: 2020-03-16T02:35:29.000Z (almost 5 years ago)
• Default Branch: master
• Last Pushed: 2020-03-26T07:25:09.000Z (almost 5 years ago)
• Last Synced: 2024-09-26T09:18:04.516Z (3 months ago)
• Topics: python
• Language: Python
• Homepage:
• Size: 58.6 KB
• Stars: 9
• Watchers: 1
• Forks: 2
• Open Issues: 1
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
# OptCAT











OptCAT (= [Optuna][1] + [CatBoost][2]) provides a scikit-learn compatible estimator that tunes hyperparameters in CatBoost with Optuna.

This Repository is very influenced by [Y-oHr-N/OptGBM](https://github.com/Y-oHr-N/OptGBM).

## Examples

```python:classification.py

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)

```

## Installation

```

pip install optcat

```

## Testing

```

poetry run pytest

```

[1]: https://optuna.org/

[2]: https://catboost.ai/