spark mllib 特征学习笔记 (二)

当然,请继续介绍其他特征处理方法的公式、适用场景和案例:

10. StringIndexer

公式

将字符串类型的标签转换为数值索引:
StringIndexer ( x ) = { 0 , 1 , 2 , ... , N − 1 } \text{StringIndexer}(x) = \{0, 1, 2, \ldots, N-1\} StringIndexer(x)={0,1,2,...,N−1}

适用场景

用于将分类标签转换为数值标签,以便机器学习算法处理。

案例

python 复制代码
from pyspark.ml.feature import StringIndexer

data = spark.createDataFrame([(0, "a"), (1, "b"), (2, "c"), (3, "a"), (4, "a"), (5, "c")], ["id", "category"])
indexer = StringIndexer(inputCol="category", outputCol="categoryIndex")
indexedData = indexer.fit(data).transform(data)
indexedData.show()

11. VectorAssembler

公式

将多个特征列合并成一个特征向量:
VectorAssembler ( x 1 , x 2 , ... , x n ) = [ x 1 , x 2 , ... , x n ] \text{VectorAssembler}(\mathbf{x}_1, \mathbf{x}_2, \ldots, \mathbf{x}_n) = [\mathbf{x}_1, \mathbf{x}_2, \ldots, \mathbf{x}_n] VectorAssembler(x1,x2,...,xn)=[x1,x2,...,xn]

适用场景

用于将多个特征列合并成一个特征向量,作为机器学习算法的输入。

案例

python 复制代码
from pyspark.ml.feature import VectorAssembler
from pyspark.ml.linalg import Vectors

data = spark.createDataFrame([(1, 2, 3), (4, 5, 6)], ["a", "b", "c"])
assembler = VectorAssembler(inputCols=["a", "b", "c"], outputCol="features")
assembledData = assembler.transform(data)
assembledData.show()

12. Word2Vec

公式

Word2Vec 是一种词嵌入模型,通过训练将单词映射到低维向量空间:
Word2Vec ( sentence ) = v \text{Word2Vec}(\text{sentence}) = \mathbf{v} Word2Vec(sentence)=v

其中 (\mathbf{v}) 是单词的向量表示。

适用场景

用于自然语言处理中的词语表示学习,以便后续应用于文本分类、文本相似度等任务。

案例

python 复制代码
from pyspark.ml.feature import Word2Vec

data = spark.createDataFrame([
    ("Hi I heard about Spark".split(" "),),
    ("I wish Java could use case classes".split(" "),),
    ("Logistic regression models are neat".split(" "),)
], ["text"])

word2Vec = Word2Vec(vectorSize=3, minCount=0, inputCol="text", outputCol="result")
model = word2Vec.fit(data)
result = model.transform(data)
result.show(truncate=False)

这些例子展示了 PySpark MLlib 中几种常用的特征处理方法的基本用法和应用场景。根据具体的数据和任务需求,选择合适的特征处理方法可以有效地提高模型的性能和准确性。

当然,请继续介绍其他特征处理方法的公式、适用场景和案例:

13. UnivariateFeatureSelector

公式

基于单变量统计测试选择特征,例如卡方检验:
UnivariateFeatureSelector ( X , y ) = { features with highest score } \text{UnivariateFeatureSelector}(X, y) = \{ \text{features with highest score} \} UnivariateFeatureSelector(X,y)={features with highest score}

适用场景

用于基于单变量统计测试(如卡方检验)选择与标签相关性最高的特征。

案例

python 复制代码
from pyspark.ml.feature import UnivariateFeatureSelector
from pyspark.ml.linalg import Vectors

data = [(1, Vectors.dense(0.0, 1.1, 0.1)),
        (0, Vectors.dense(2.0, 1.0, -1.0)),
        (0, Vectors.dense(2.0, 1.3, 1.0)),
        (1, Vectors.dense(0.0, 1.2, -0.5))]
df = spark.createDataFrame(data, ["label", "features"])

selector = UnivariateFeatureSelector(featuresCol="features", outputCol="selectedFeatures", labelCol="label", selectionMode="numTopFeatures", numTopFeatures=1)
result = selector.fit(df).transform(df)
result.show()

14. VarianceThresholdSelector

公式

基于方差选择特征,移除方差低于阈值的特征:
VarianceThresholdSelector ( X ) = { features with variance above threshold } \text{VarianceThresholdSelector}(X) = \{ \text{features with variance above threshold} \} VarianceThresholdSelector(X)={features with variance above threshold}

适用场景

用于移除方差较低的特征,以减少噪声对模型的影响。

案例

python 复制代码
from pyspark.ml.feature import VarianceThresholdSelector
from pyspark.ml.linalg import Vectors

data = [(1, Vectors.dense(0.0, 1.0, 0.0)),
        (0, Vectors.dense(0.0, 1.0, 1.0)),
        (0, Vectors.dense(0.0, 1.0, 0.0))]
df = spark.createDataFrame(data, ["label", "features"])

selector = VarianceThresholdSelector(featuresCol="features", outputCol="selectedFeatures", threshold=0.0)
result = selector.fit(df).transform(df)
result.show()

15. VectorIndexer

公式

索引化向量中的类别特征列:
VectorIndexer ( X ) = { indexed features } \text{VectorIndexer}(X) = \{ \text{indexed features} \} VectorIndexer(X)={indexed features}

适用场景

用于处理数据集中的向量特征,自动识别并索引类别特征。

案例

python 复制代码
from pyspark.ml.feature import VectorIndexer
from pyspark.ml.linalg import Vectors

data = [(Vectors.dense(1.0, 2.0, 3.0),),
        (Vectors.dense(2.0, 5.0, 6.0),),
        (Vectors.dense(1.0, 8.0, 9.0),)]
df = spark.createDataFrame(data, ["features"])

indexer = VectorIndexer(inputCol="features", outputCol="indexed", maxCategories=2)
indexedData = indexer.fit(df).transform(df)
indexedData.show()

16. VectorSizeHint

公式

添加向量列的大小信息到元数据中:
VectorSizeHint ( X ) = { features with size hint in metadata } \text{VectorSizeHint}(X) = \{ \text{features with size hint in metadata} \} VectorSizeHint(X)={features with size hint in metadata}

适用场景

用于在向量列中添加大小信息,以提供给后续流水线阶段使用。

案例

python 复制代码
from pyspark.ml.feature import VectorSizeHint
from pyspark.ml.linalg import Vectors

data = [(Vectors.dense([1.0, 2.0]),),
        (Vectors.dense([2.0, 3.0]),)]
df = spark.createDataFrame(data, ["features"])

sizeHint = VectorSizeHint(inputCol="features", size=2)
sizeHint.transform(df).show()

这些例子展示了更多 PySpark MLlib 中特征处理方法的公式、适用场景和简单案例。每种方法都有其特定的数学原理和适用范围,根据具体任务的需求选择合适的方法可以提高数据处理的效率和模型的预测性能。

相关推荐
大丈夫立于天地间23 分钟前
ISIS基础知识
网络·网络协议·学习·智能路由器·信息与通信
doubt。31 分钟前
【BUUCTF】[RCTF2015]EasySQL1
网络·数据库·笔记·mysql·安全·web安全
Chambor_mak1 小时前
stm32单片机个人学习笔记14(USART串口数据包)
stm32·单片机·学习
Zelotz1 小时前
线段树与矩阵
笔记
汇能感知2 小时前
光谱相机在智能冰箱的应用原理与优势
经验分享·笔记·科技
PaLu-LI2 小时前
ORB-SLAM2源码学习:Initializer.cc⑧: Initializer::CheckRT检验三角化结果
c++·人工智能·opencv·学习·ubuntu·计算机视觉
yuanbenshidiaos2 小时前
【大数据】机器学习----------计算机学习理论
大数据·学习·机器学习
汤姆和佩琦2 小时前
2025-1-20-sklearn学习(42) 使用scikit-learn计算 钿车罗帕,相逢处,自有暗尘随马。
人工智能·python·学习·机器学习·scikit-learn·sklearn
Tech智汇站3 小时前
Quick Startup,快捷处理自启程序的工具,加快电脑开机速度!
经验分享·科技·学习·学习方法·改行学it
qq_312738453 小时前
jvm学习总结
jvm·学习