Python pyspark MLUtils.convertVectorColumnsToML用法及代碼示例

本文簡要介紹 pyspark.mllib.util.MLUtils.convertVectorColumnsToML 的用法。

用法: static convertVectorColumnsToML(dataset, *cols)

將輸入 DataFrame 中的向量列從 pyspark.mllib.linalg.Vector 類型轉換為 spark.ml 包下的新 pyspark.ml.linalg.Vector 類型。

2.0.0 版中的新函數。

參數：

dataset：DataFrame

輸入數據集

*cols：str

要轉換的向量列。

新的向量列將被忽略。如果未指定，則所有舊向量列都將轉換為嵌套的除外。

DataFrame: 將舊向量列轉換為新向量類型的輸入數據集

例子：

>>> import pyspark
>>> from pyspark.mllib.linalg import Vectors
>>> from pyspark.mllib.util import MLUtils
>>> df = spark.createDataFrame(
...     [(0, Vectors.sparse(2, [1], [1.0]), Vectors.dense(2.0, 3.0))],
...     ["id", "x", "y"])
>>> r1 = MLUtils.convertVectorColumnsToML(df).first()
>>> isinstance(r1.x, pyspark.ml.linalg.SparseVector)
True
>>> isinstance(r1.y, pyspark.ml.linalg.DenseVector)
True
>>> r2 = MLUtils.convertVectorColumnsToML(df, "x").first()
>>> isinstance(r2.x, pyspark.ml.linalg.SparseVector)
True
>>> isinstance(r2.y, pyspark.mllib.linalg.DenseVector)
True

相關用法

注：本文由純淨天空篩選整理自spark.apache.org大神的英文原創作品 pyspark.mllib.util.MLUtils.convertVectorColumnsToML。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。

用法:

參數：

返回：

例子：