ClickHouse/docs/en/guides/apply_catboost_model.md

222 lines
6.4 KiB
Markdown
Raw Normal View History

2019-08-22 11:34:56 +00:00
# Applying a Catboost Model in ClickHouse {#applying-catboost-model-in-clickhouse}
2019-10-10 12:17:39 +00:00
[CatBoost](https://catboost.ai) is a free and open-source gradient boosting library developed at [Yandex](https://yandex.com/company/) for machine learning.
2019-08-22 11:34:56 +00:00
With this instruction, you will learn to apply pre-trained models in ClickHouse: as a result, you run the model inference from SQL.
To apply a CatBoost model in ClickHouse:
2019-08-22 11:34:56 +00:00
1. [Create a Table](#create-table).
2019-10-10 12:29:06 +00:00
2. [Insert the Data to the Table](#insert-data-to-table).
3. [Build libcatboostmodel and Configure the Model](#build-libcatboostmodel-and-configure-model) (Optional step).
2019-10-10 12:29:06 +00:00
4. [Run the Model Inference from SQL](#run-model-inference).
2019-08-22 05:45:45 +00:00
For more information about training CatBoost models, see [Training and applying models](https://catboost.ai/docs/features/training.html#training).
2019-08-22 11:34:56 +00:00
## Prerequisites {#prerequisites}
If you don't have the [Docker](https://docs.docker.com/install/) yet, install it.
2019-08-22 11:34:56 +00:00
!!! note "Note"
[Docker](https://www.docker.com) is a software platform that allows you to create containers that isolate a CatBoost and ClickHouse installation from the rest of the system.
Before applying a CatBoost model:
**1.** Pull the [Docker image](https://hub.docker.com/r/yandex/tutorial-catboost-clickhouse) from the registry:
```bash
$ docker pull yandex/tutorial-catboost-clickhouse
```
2019-08-22 11:34:56 +00:00
This Docker image contains everything you need to run CatBoost and ClickHouse: code, runtime, libraries, environment variables, and configuration files.
2019-08-22 15:13:36 +00:00
**2.** Make sure the Docker image has been successfully pulled:
```bash
$ docker image ls
REPOSITORY TAG IMAGE ID CREATED SIZE
2019-10-10 12:17:39 +00:00
yandex/tutorial-catboost-clickhouse latest 622e4d17945b 22 hours ago 1.37GB
```
2019-08-22 15:13:36 +00:00
**3.** Start a Docker container based on this image:
```bash
$ docker run -it -p 8888:8888 yandex/tutorial-catboost-clickhouse
```
2019-08-22 11:34:56 +00:00
## 1. Create a Table {#create-table}
To create a ClickHouse table for the train sample:
**1.** Start ClickHouse console client in interactive mode:
```bash
$ clickhouse client
```
2019-08-22 11:34:56 +00:00
!!! note "Note"
The ClickHouse server is already running inside the Docker container.
**2.** Create the table using the command:
```sql
:) CREATE TABLE amazon_train
(
date Date MATERIALIZED today(),
ACTION UInt8,
RESOURCE UInt32,
MGR_ID UInt32,
ROLE_ROLLUP_1 UInt32,
ROLE_ROLLUP_2 UInt32,
ROLE_DEPTNAME UInt32,
ROLE_TITLE UInt32,
ROLE_FAMILY_DESC UInt32,
ROLE_FAMILY UInt32,
ROLE_CODE UInt32
)
2019-10-10 12:17:39 +00:00
ENGINE = MergeTree()
```
**3.** Exit from ClickHouse console client:
```sql
:) exit
```
2019-10-10 12:17:39 +00:00
## 2. Insert the Data to the Table {#insert-data-to-table}
To insert the data:
**1.** Run the following command:
```bash
$ clickhouse client --host 127.0.0.1 --query 'INSERT INTO amazon_train FORMAT CSVWithNames' < ~/amazon/train.csv
```
**2.** Start ClickHouse console client in interactive mode:
```bash
$ clickhouse client
```
**3.** Make sure the data has been uploaded:
```sql
:) SELECT count() FROM amazon_train
2019-10-10 12:17:39 +00:00
SELECT count()
FROM amazon_train
+-count()-+
2019-10-10 12:17:39 +00:00
| 65538 |
+---------+
```
## 3. Build libcatboostmodel Configure the Model to Work with the Trained Model {#build-libcatboostmodel-and-configure-model}
!!! note "Note"
**Optional step.** The Docker image contains the library `.data/libcatboostmodel.so` and model configuration file `models/amazon_model.xml`.
The library `libcatboostmodel.<so|dll|dylib>` is the CatBoost model interface library. To build the library see [CatBoost documentation](https://catboost.ai/docs/concepts/c-plus-plus-api_dynamic-c-pluplus-wrapper.html).
To configure the model:
2019-10-10 12:17:39 +00:00
**1.** Create a model config file in the `models` folder (for example, `models/config_model.xml`):
```xml
<models>
<model>
<!-- Model type. Now catboost only. -->
<type>catboost</type>
<!-- Model name. -->
<name>amazon</name>
<!-- Path to trained model. -->
<path>/home/catboost/tutorial/catboost_model.bin</path>
<!-- Update interval. -->
<lifetime>0</lifetime>
</model>
</models>
```
2019-10-10 12:17:39 +00:00
**2.** Add a link to the created file to the ClickHouse configuration `../../etc/clickhouse-server/config.d/models_config.xml`.
```xml
<models_config>/home/catboost/models/*_model.xml</models_config>
```
2019-10-10 12:17:39 +00:00
The ClickHouse config file should already have this setting. To check it, run `tail ../../etc/clickhouse-server/config.d/models_config.xml`.
2019-10-10 12:17:39 +00:00
## 4. Run the Model Inference from SQL {#run-model-inference}
2019-10-10 12:17:39 +00:00
For test model run the ClickHouse client `$ clickhouse client`.
2019-08-22 11:34:56 +00:00
Let's make sure that the model is working:
```sql
:) SELECT
modelEvaluate('amazon',
RESOURCE,
MGR_ID,
ROLE_ROLLUP_1,
ROLE_ROLLUP_2,
ROLE_DEPTNAME,
ROLE_TITLE,
ROLE_FAMILY_DESC,
ROLE_FAMILY,
ROLE_CODE) > 0 AS prediction,
ACTION AS target
FROM amazon_train
LIMIT 10
```
2019-08-22 11:34:56 +00:00
!!! note "Note"
Function [modelEvaluate](../query_language/functions/other_functions.md#function-modelevaluate) returns tuple with per-class raw predictions for multiclass models.
2019-08-22 11:34:56 +00:00
Let's predict probability:
```sql
:) SELECT
modelEvaluate('amazon',
RESOURCE,
MGR_ID,
ROLE_ROLLUP_1,
ROLE_ROLLUP_2,
ROLE_DEPTNAME,
ROLE_TITLE,
ROLE_FAMILY_DESC,
ROLE_FAMILY,
ROLE_CODE) AS prediction,
1. / (1 + exp(-prediction)) AS probability,
ACTION AS target
FROM amazon_train
LIMIT 10
```
2019-08-22 11:34:56 +00:00
!!! note "Note"
More info about [exp()](../query_language/functions/math_functions.md) function.
Let's calculate LogLoss on the sample:
```sql
:) SELECT -avg(tg * log(prob) + (1 - tg) * log(1 - prob)) AS logloss
FROM
(
SELECT
modelEvaluate('amazon',
RESOURCE,
MGR_ID,
ROLE_ROLLUP_1,
ROLE_ROLLUP_2,
ROLE_DEPTNAME,
ROLE_TITLE,
ROLE_FAMILY_DESC,
ROLE_FAMILY,
ROLE_CODE) AS prediction,
1. / (1. + exp(-prediction)) AS prob,
ACTION AS tg
FROM amazon_train
)
2019-08-22 11:34:56 +00:00
```
!!! note "Note"
More info about [avg()](../query_language/agg_functions/reference.md#agg_function-avg) and [log()](../query_language/functions/math_functions.md) functions.