BigQueryML：随机森林分类

Question

你好 stackoverflow 社区！

我最近发现 Bigquery ML 不支持随机森林分类模型。为了克服这个问题，我认为我可以使用 sklearn 包构建一个模型，然后在 bigqueryml 上使用相同的超参数。在这种情况下，在 bigquery 中使用提升树模型是最佳选择吗？

Answer 1

您可以使用 BQML 的提升树模型类型训练随机森林模型，并使用以下参数

1. 将 BOOSTER_TYPE 设置为 GBTREE
2. 将 NUM_PARALLEL_TREE 设置为大于 1 的某个数字
3. 将 MAX_ITERATIONS 设置为 1
4. 将 LEARN_RATE 设置为 1
5. 将 {COLSAMPLE_BYTREE、COLSAMPLE_BYLEVEL、COLSAMPLE_BYNODE} 之一设置为小于1.

Answer 2

BQ ML 中添加了随机森林模型。

CREATE MODEL mydataset.model1
OPTIONS(MODEL_TYPE = 'RANDOM_FOREST_CLASSIFIER')
AS SELECT * FROM mydataset.mytable;

文档 - https:// cloud.google.com/bigquery-ml/docs/reference/standard-sql/bigqueryml-syntax-create-random-forest