Java 类名：com.alibaba.alink.operator.batch.clustering.KMeansPredictBatchOp
Python 类名：KMeansPredictBatchOp

功能介绍

KMeans 是一个经典的聚类算法。
基本思想是：以空间中k个点为中心进行聚类，对最靠近他们的对象归类。通过迭代的方法，逐次更新各聚类中心的值，直至得到最好的聚类结果。
Alink上KMeans算法包括KMeans，KMeans批量预测, KMeans流式预测。

参数说明

| 名称 | 中文名称 | 描述 | 类型 | 是否必须？ | 取值范围 | 默认值 | | —- | —- | —- | —- | —- | —- | —- |

| predictionCol | 预测结果列名 | 预测结果列名 | String | ✓ | | |

| modelFilePath | 模型的文件路径 | 模型的文件路径 | String | | | null |

| predictionDetailCol | 预测详细信息列名 | 预测详细信息列名 | String | | | |

| predictionDistanceCol | 预测距离列名 | 预测距离列名 | String | | | |

| reservedCols | 算法保留列名 | 算法保留列 | String[] | | | null |

| numThreads | 组件多线程线程个数 | 组件多线程线程个数 | Integer | | | 1 |

代码示例

Python 代码

  1. from pyalink.alink import *
  3. import pandas as pd
  5. useLocalEnv(1)
  7. df = pd.DataFrame([
  8.     [0, "0 0 0"],
  9.     [1, "0.1,0.1,0.1"],
  10.     [2, "0.2,0.2,0.2"],
  11.     [3, "9 9 9"],
  12.     [4, "9.1 9.1 9.1"],
  13.     [5, "9.2 9.2 9.2"]
  14. ])
  16. inOp1 = BatchOperator.fromDataframe(df, schemaStr='id int, vec string')
  17. inOp2 = StreamOperator.fromDataframe(df, schemaStr='id int, vec string')
  19. kmeans = KMeansTrainBatchOp()\
  20.     .setVectorCol("vec")\
  21.     .setK(2)\
  22.     .linkFrom(inOp1)
  23. kmeans.lazyPrint(10)
  25. predictBatch = KMeansPredictBatchOp()\
  26.     .setPredictionCol("pred")\
  27.     .linkFrom(kmeans, inOp1)
  28. predictBatch.print()
  30. predictStream = KMeansPredictStreamOp(kmeans)\
  31.     .setPredictionCol("pred")\
  32.     .linkFrom(inOp2)
  33. predictStream.print()
  35. StreamOperator.execute()

Java 代码

  1. import org.apache.flink.types.Row;
  3. import com.alibaba.alink.operator.batch.BatchOperator;
  4. import com.alibaba.alink.operator.batch.clustering.KMeansPredictBatchOp;
  5. import com.alibaba.alink.operator.batch.clustering.KMeansTrainBatchOp;
  6. import com.alibaba.alink.operator.batch.source.MemSourceBatchOp;
  7. import com.alibaba.alink.operator.stream.StreamOperator;
  8. import com.alibaba.alink.operator.stream.clustering.KMeansPredictStreamOp;
  9. import com.alibaba.alink.operator.stream.source.MemSourceStreamOp;
  10. import org.junit.Test;
  12. import java.util.Arrays;
  13. import java.util.List;
  15. public class KMeansPredictBatchOpTest {
  16.     @Test
  17.     public void testKMeansPredictBatchOp() throws Exception {
  18.         List <Row> df = Arrays.asList(
  19.             Row.of(0, "0 0 0"),
  20.             Row.of(1, "0.1,0.1,0.1"),
  21.             Row.of(2, "0.2,0.2,0.2"),
  22.             Row.of(3, "9 9 9"),
  23.             Row.of(4, "9.1 9.1 9.1"),
  24.             Row.of(5, "9.2 9.2 9.2")
  25.         );
  26.         BatchOperator <?> inOp1 = new MemSourceBatchOp(df, "id int, vec string");
  27.         StreamOperator <?> inOp2 = new MemSourceStreamOp(df, "id int, vec string");
  28.         BatchOperator <?> kmeans = new KMeansTrainBatchOp()
  29.             .setVectorCol("vec")
  30.             .setK(2)
  31.             .linkFrom(inOp1);
  32.         kmeans.lazyPrint(10);
  34.         BatchOperator <?> predictBatch = new KMeansPredictBatchOp()
  35.             .setPredictionCol("pred")
  36.             .linkFrom(kmeans, inOp1);
  37.         predictBatch.print();
  39.         StreamOperator <?> predictStream = new KMeansPredictStreamOp(kmeans)
  40.             .setPredictionCol("pred")
  41.             .linkFrom(inOp2);
  42.         predictStream.print();
  43.         StreamOperator.execute();
  44.     }
  45. }

运行结果

模型结果

| model_id | model_info | | —- | —- |

| 0 | {“vectorCol”:””vec””,”latitudeCol”:null,”longitudeCol”:null,”distanceType”:””EUCLIDEAN””,”k”:”2”,”vectorSize”:”3”} |

| 1048576 | {“clusterId”:0,”weight”:3.0,”vec”:{“data”:[9.099999999999998,9.099999999999998,9.099999999999998]}} |

| 2097152 | {“clusterId”:1,”weight”:3.0,”vec”:{“data”:[0.1,0.1,0.1]}} |

预测结果

| id | vec | pred | | —- | —- | —- |

| 0 | 0 0 0 | 1 |

| 1 | 0.1,0.1,0.1 | 1 |

| 2 | 0.2,0.2,0.2 | 1 |

| 3 | 9 9 9 | 0 |

| 4 | 9.1 9.1 9.1 | 0 |

| 5 | 9.2 9.2 9.2 | 0 |