Spearman
模型接口类别 |
函数接口 |
|---|---|
ML API |
def corr(dataset: Dataset[_], column: String, method: String): DataFrame |
- 输入输出
- 包名：org.apache.spark.ml.stat
- 类名：Correlation
- 方法名：corr
- 输入：Dataset[_]，训练样本数据，必须字段如下：
Param name
Type(s)
Description
data
Dataset[Vector]
矩阵，以行为单位进行存储
column
String
指定进行列进行相关矩阵计算
method
String
相关矩阵方法，可以选择spearman
- 算法参数
Param name
Type(s)
Default
Description
method
String
pearson
求解相关矩阵的方法
代码接口示例：
1val mat_df = ml.stat.Correlation.corr(data,"matrix", method = "spearman")
- 输出：Pearson相关矩阵Matrix：
Param name
Type(s)
Description
df
DataFrame
Spearman相关矩阵，列名为column+method
- 使用样例
val mat = stat.Correlation.corr(data, "matrix", “spearman”)
父主题： 算法API