ClickHouse/docs/zh/guides/apply-catboost-model.md

240 lines
6.6 KiB
Markdown
Raw Normal View History

2020-04-03 13:23:32 +00:00
---
machine_translated: true
machine_translated_rev: b111334d6614a02564cf32f379679e9ff970d9b1
toc_priority: 41
toc_title: "\u5E94\u7528CatBoost\u6A21\u578B"
2020-04-03 13:23:32 +00:00
---
# 在ClickHouse中应用Catboost模型 {#applying-catboost-model-in-clickhouse}
2020-04-03 13:23:32 +00:00
[CatBoost](https://catboost.ai) 是一个自由和开源的梯度提升库开发 [Yandex](https://yandex.com/company/) 用于机器学习。
2020-04-03 13:23:32 +00:00
通过此指令您将学习如何通过从SQL运行模型推理在ClickHouse中应用预先训练好的模型。
2020-04-03 13:23:32 +00:00
在ClickHouse中应用CatBoost模型:
2020-04-03 13:23:32 +00:00
1. [创建表](#create-table).
2. [将数据插入到表中](#insert-data-to-table).
3. [碌莽禄into拢Integrate010-68520682\<url\>](#integrate-catboost-into-clickhouse) (可选步骤)。
4. [从SQL运行模型推理](#run-model-inference).
2020-04-03 13:23:32 +00:00
有关训练CatBoost模型的详细信息请参阅 [培训和应用模型](https://catboost.ai/docs/features/training.html#training).
2020-04-03 13:23:32 +00:00
## 先决条件 {#prerequisites}
2020-04-03 13:23:32 +00:00
如果你没有 [Docker](https://docs.docker.com/install/) 然而,安装它。
2020-04-03 13:23:32 +00:00
!!! note "注"
[Docker](https://www.docker.com) 是一个软件平台允许您创建容器将CatBoost和ClickHouse安装与系统的其余部分隔离。
2020-04-03 13:23:32 +00:00
在应用CatBoost模型之前:
2020-04-03 13:23:32 +00:00
**1.** 拉 [码头窗口映像](https://hub.docker.com/r/yandex/tutorial-catboost-clickhouse) 从注册表:
2020-04-03 13:23:32 +00:00
``` bash
$ docker pull yandex/tutorial-catboost-clickhouse
```
此Docker映像包含运行CatBoost和ClickHouse所需的所有内容代码、运行时、库、环境变量和配置文件。
2020-04-03 13:23:32 +00:00
**2.** 确保已成功拉取Docker映像:
2020-04-03 13:23:32 +00:00
``` bash
$ docker image ls
REPOSITORY TAG IMAGE ID CREATED SIZE
yandex/tutorial-catboost-clickhouse latest 622e4d17945b 22 hours ago 1.37GB
```
**3.** 基于此映像启动一个Docker容器:
2020-04-03 13:23:32 +00:00
``` bash
$ docker run -it -p 8888:8888 yandex/tutorial-catboost-clickhouse
```
## 1. 创建表 {#create-table}
2020-04-03 13:23:32 +00:00
为训练样本创建ClickHouse表:
2020-04-03 13:23:32 +00:00
**1.** 在交互模式下启动ClickHouse控制台客户端:
2020-04-03 13:23:32 +00:00
``` bash
$ clickhouse client
```
!!! note "注"
ClickHouse服务器已经在Docker容器内运行。
2020-04-03 13:23:32 +00:00
**2.** 使用以下命令创建表:
2020-04-03 13:23:32 +00:00
``` 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
)
ENGINE = MergeTree ORDER BY date
```
**3.** 从ClickHouse控制台客户端退出:
2020-04-03 13:23:32 +00:00
``` sql
:) exit
```
## 2. 将数据插入到表中 {#insert-data-to-table}
2020-04-03 13:23:32 +00:00
插入数据:
2020-04-03 13:23:32 +00:00
**1.** 运行以下命令:
2020-04-03 13:23:32 +00:00
``` bash
$ clickhouse client --host 127.0.0.1 --query 'INSERT INTO amazon_train FORMAT CSVWithNames' < ~/amazon/train.csv
```
**2.** 在交互模式下启动ClickHouse控制台客户端:
2020-04-03 13:23:32 +00:00
``` bash
$ clickhouse client
```
**3.** 确保数据已上传:
2020-04-03 13:23:32 +00:00
``` sql
:) SELECT count() FROM amazon_train
SELECT count()
FROM amazon_train
+-count()-+
| 65538 |
+-------+
2020-04-03 13:23:32 +00:00
```
## 3. 碌莽禄into拢Integrate010-68520682\<url\> {#integrate-catboost-into-clickhouse}
2020-04-03 13:23:32 +00:00
!!! note "注"
**可选步骤。** Docker映像包含运行CatBoost和ClickHouse所需的所有内容。
2020-04-03 13:23:32 +00:00
碌莽禄to拢integrate010-68520682\<url\>:
2020-04-03 13:23:32 +00:00
**1.** 构建评估库。
2020-04-03 13:23:32 +00:00
评估CatBoost模型的最快方法是编译 `libcatboostmodel.<so|dll|dylib>` 图书馆. 有关如何构建库的详细信息,请参阅 [CatBoost文件](https://catboost.ai/docs/concepts/c-plus-plus-api_dynamic-c-pluplus-wrapper.html).
2020-04-03 13:23:32 +00:00
**2.** 例如,在任何地方和任何名称创建一个新目录, `data` 并将创建的库放入其中。 Docker映像已经包含了库 `data/libcatboostmodel.so`.
2020-04-03 13:23:32 +00:00
**3.** 例如在任何地方和任何名称为config model创建一个新目录, `models`.
2020-04-03 13:23:32 +00:00
**4.** 创建具有任意名称的模型配置文件,例如, `models/amazon_model.xml`.
2020-04-03 13:23:32 +00:00
**5.** 描述模型配置:
2020-04-03 13:23:32 +00:00
``` 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>
```
**6.** 将CatBoost的路径和模型配置添加到ClickHouse配置:
2020-04-03 13:23:32 +00:00
``` xml
<!-- File etc/clickhouse-server/config.d/models_config.xml. -->
<catboost_dynamic_library_path>/home/catboost/data/libcatboostmodel.so</catboost_dynamic_library_path>
<models_config>/home/catboost/models/*_model.xml</models_config>
```
## 4. 从SQL运行模型推理 {#run-model-inference}
2020-04-03 13:23:32 +00:00
对于测试模型运行ClickHouse客户端 `$ clickhouse client`.
2020-04-03 13:23:32 +00:00
让我们确保模型正常工作:
2020-04-03 13:23:32 +00:00
``` 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
```
!!! note "注"
功能 [模型值](../sql-reference/functions/other-functions.md#function-modelevaluate) 返回带有多类模型的每类原始预测的元组。
2020-04-03 13:23:32 +00:00
让我们预测一下:
2020-04-03 13:23:32 +00:00
``` 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
```
!!! note "注"
更多信息 [exp()](../sql-reference/functions/math-functions.md) 功能。
2020-04-03 13:23:32 +00:00
让我们计算样本的LogLoss:
2020-04-03 13:23:32 +00:00
``` 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
)
```
!!! note "注"
更多信息 [avg()](../sql-reference/aggregate-functions/reference.md#agg_function-avg) 和 [日志()](../sql-reference/functions/math-functions.md) 功能。
2020-04-03 13:23:32 +00:00
[原始文章](https://clickhouse.tech/docs/en/guides/apply_catboost_model/) <!--hide-->