如何获取BigQuery中Kmeans聚类表自动生成的聚类数量

你创建模型时没有在OPTIONS中显式指定num_clusters参数，BigQuery会自动推断最优聚类数量，要获取已训练好的ch09eu.london_station_clusters模型的实际聚类数，可通过以下两种方式实现：

• 方法1：调用ML.CENTROIDS函数统计质心数量

SELECT COUNT(DISTINCT centroid_id) AS cluster_count
FROM ML.CENTROIDS(MODEL `ch09eu.london_station_clusters`)

查询返回的cluster_count字段值就是最终的聚类数量。

• 方法2：查询模型元数据信息表

SELECT
  option_value AS cluster_count
FROM
  `ch09eu.INFORMATION_SCHEMA.MODEL_OPTIONS`
WHERE
  model_name = 'london_station_clusters'
  AND option_name = 'num_clusters'

如果你后续创建K-Means模型不需要自动推断聚类数量，直接在OPTIONS参数中添加num_clusters = 自定义数值即可固定聚类数。

内容的提问来源于stack exchange，提问作者user16788731