功能介绍

主成分分析,是考察多个变量间相关性一种多元统计方法,研究如何通过少数几个主成分来揭示多个变量间的内部结构,即从原始变量中导出少数几个主成分,使它们尽可能多地保留原始变量的信息,且彼此间互不相关,作为新的综合指标。详细介绍请见维基百科链接wiki

参数说明

名称 中文名称 描述 类型 是否必须? 默认值
k 降维后的维度 降维后的维度 Integer
calculationType 计算类型 计算类型,包含”CORR”, “COV_SAMPLE”, “COVAR_POP”三种。 String “CORR”
selectedCols 选中的列名数组 计算列对应的列名列表 String[] null
vectorCol 向量列名 向量列对应的列名,默认值是null String null
withMean 是否使用均值 是否使用均值,默认使用 Boolean true
withStd 是否使用标准差 是否使用标准差,默认使用 Boolean true

脚本示例

脚本

  1. data = np.array([
  2. [0.0,0.0,0.0],
  3. [0.1,0.2,0.1],
  4. [0.2,0.2,0.8],
  5. [9.0,9.5,9.7],
  6. [9.1,9.1,9.6],
  7. [9.2,9.3,9.9]
  8. ])
  9. df = pd.DataFrame({"x1": data[:, 0], "x2": data[:, 1], "x3": data[:, 2]})
  10. # batch source
  11. inOp = dataframeToOperator(df, schemaStr='x1 double, x2 double, x3 double', op_type='batch')
  12. trainOp = PcaTrainBatchOp()\
  13. .setK(2)\
  14. .setSelectedCols(["x1","x2","x3"])
  15. predictOp = PcaPredictBatchOp()\
  16. .setPredictionCol("pred")
  17. # batch train
  18. inOp.link(trainOp)
  19. # batch predict
  20. predictOp.linkFrom(trainOp,inOp)
  21. predictOp.print()
  22. # stream predict
  23. inStreamOp = dataframeToOperator(df, schemaStr='x1 double, x2 double, x3 double', op_type='stream')
  24. predictStreamOp = PcaPredictStreamOp(trainOp)\
  25. .setPredictionCol("pred")
  26. predictStreamOp.linkFrom(inStreamOp)
  27. predictStreamOp.print()
  28. StreamOperator.execute()

结果

x1 x2 x3 pred
9.0 9.5 9.7 3.2280384305400736,1.1516225426477789E-4
0.2 0.2 0.8 0.13565076707329407,0.09003329494282108
9.2 9.3 9.9 3.250783163664603,0.0456526246528135
9.1 9.1 9.6 3.182618319978973,0.027469531992220464
0.1 0.2 0.1 0.045855205015063565,-0.012182917696915518
0.0 0.0 0.0 0.0,0.0