【机器学习】Spark ML 对数据进行规范化预处理 StandardScaler 与向量拆分

本文涉及的产品
实时数仓Hologres,5000CU*H 100GB 3个月
实时计算 Flink 版,5000CU*H 3个月
智能开放搜索 OpenSearch行业算法版,1GB 20LCU 1个月
简介: 标准化Scaler是数据预处理技术,用于将特征值映射到均值0、方差1的标准正态分布,以消除不同尺度特征的影响,提升模型稳定性和精度。Spark ML中的StandardScaler实现此功能,通过`.setInputCol`、`.setOutputCol`等方法配置并应用到DataFrame数据。示例展示了如何在Spark中使用StandardScaler进行数据规范化,包括创建SparkSession,构建DataFrame,使用VectorAssembler和StandardScaler,以及将向量拆分为列。规范化有助于降低特征重要性,提高模型训练速度和计算效率。

什么数据规范化?

规范化(Normalization)是一种数据预处理技术,用于将不同范围的特征值映射到相同的范围内。其中,StandardScaler 是一种规范化的方法,它将特征值转换为均值为 0、方差为 1 的标准正态分布。

需要注意的是,StandardScaler 的使用需要对数据进行归一化,以确保每个特征值的范围相同。否则,具有较大范围的特征值可能会在标准化后对结果产生较大的影响。

为什么要对数据进行规范化处理?

对数据进行规范化处理有以下几个原因:

  1. 降低特征的重要性:在机器学习中,不同的特征通常具有不同的尺度和范围,这可能会导致某些特征在预测中具有较大的重要性。通过规范化处理,可以将所有特征缩放到相同的尺度和范围,降低特征的重要性,从而提高模型的稳定性和可靠性。

  2. 提高模型的精度:在某些机器学习算法中,例如K均值聚类和支持向量机,距离和相似度的度量对算法的结果产生很大影响。如果特征具有不同的尺度和范围,可能会导致距离和相似度的度量产生偏差,从而降低模型的精度。通过规范化处理,可以消除这种偏差,提高模型的精度。

  3. 加速模型的训练:在机器学习中,通常需要对数据进行多次迭代来训练模型。如果特征具有不同的尺度和范围,可能会导致收敛速度较慢,从而延长训练时间。通过规范化处理,可以提高收敛速度,加速模型的训练。

  4. 优化计算资源的利用:在大数据环境中,对数据进行规范化处理可以使得机器学习算法在分布式计算环境中更好地运行,从而充分利用计算资源,提高计算效率。

进行规范化处理可以提高模型的稳定性、可靠性、精度、训练速度和计算效率,从而更好地应用于实际场景中。

Spark ML 之 StandardScaler

StandardScaler 是 Spark ML 中的一个数据转换器,用于对数据进行规范化处理。它可以将数据集的每个特征按照均值和标准差进行转换,使得所有特征具有相同的尺度和范围,从而降低特征的重要性,提高模型的稳定性和可靠性。

Spark ML 应用示例

首先,引入 Spark ML 的依赖,这里不做过多的赘述了。

        <dependency>
            <groupId>org.apache.spark</groupId>
            <artifactId>spark-mllib_2.12</artifactId>
            <version>3.0.0</version>
        </dependency>
AI 代码解读

1.创建 SparkSession 对象

        val spark: SparkSession = SparkSession
                .builder
                .master("local[*]")
                .appName("Test")
                .getOrCreate()
AI 代码解读

2.创建一个示例数据集,格式为 DataFrame

        import spark.implicits._

        val data: DataFrame = Seq(
            (1.0, 2.0, 3.0),
            (2.0, 3.0, 4.0),
            (3.0, 4.0, 5.0)
        ).toDF("feature1","feature2","feature3")
AI 代码解读

3.将 DataFrame 转换为向量

        val assembler: VectorAssembler = new VectorAssembler()
                .setInputCols(Array("feature1", "feature2", "feature3"))
                .setOutputCol("features")

        val assemblerData: DataFrame = assembler.transform(data)
AI 代码解读

将所有的列都作为输入列]("feature1", "feature2", "feature3"),转换为特征向量。

VectorAssembler 是 Spark MLlib 中的一个Transformer,其作用是将多个列组合成一个特征向量。

在机器学习中,通常需要将多个特征组合在一起作为模型的输入,而 VectorAssembler 可以很方便地完成这个任务,它可以将数据集中的多个列组合成一个稠密的向量或者一个稀疏的向量,这取决于输入的列和参数设置。

4.创建 StandardScaler 实例,并设置输入和输出列

        val scaler = new StandardScaler()
                .setInputCol("features")
                .setOutputCol("scaled_features")
                .setWithMean(true)
                .setWithStd(true)
AI 代码解读

参数说明:

  • .setInputCol("features") 设置输入列的名称为 "features",即指定要对哪个列的数据进行规范化处理。这里的 "features" 是我们在转换向量时 DataFrame 的输出列名。

  • .setOutputCol("scaled_features") 设置输出列的名称为 "scaled_features",即指定规范化处理后的数据要保存到哪个列。这个列名可以根据需要自行命名,不一定非得叫 "scaled_features"

  • .setWithMean(true) 设置是否将数据中心化,即将数据的均值设为 0。默认值为 false,即不中心化。

  • .setWithStd(true) 设置是否将数据标准化,即将数据除以标准差。默认值为 true,即标准化数据,使得数据的均值为 0,方差为 1,从而将数据转化为标准正态分布。

中心化就是将数据减去其均值,使其以 0 为中心点,也称为零均值化。具体来说,对于一个向量 X,其中心化后的向量 X' 可以表示为: X' = X - mean(X)
其中 mean(X) 是向量 X 的均值。中心化后的向量 X' 的均值将会变为 0,因为:mean(X') = mean(X - mean(X)) = mean(X) - mean(X) = 0
这样做的好处是,消除了数据的平移性,将数据转化为以 0 为中心的分布,便于不同指标之间的比较和加权。同时,中心化并不会改变数据的比例关系,也就是说,中心化后的数据与原始数据之间的关系不会发生变化,因此,中心化操作并不会影响模型的精度和稳定性,反而可以提高模型的精度和稳定性。

5.对数据集进行拟合与转换

        val scalerModel = scaler.fit(assemblerData)
        val scaledData = scalerModel.transform(assemblerData)
AI 代码解读

6.输出处理完成的数据

        scaledData.show(false)
AI 代码解读

输入结果如下所示:

+--------+--------+--------+-------------+----------------+
|feature1|feature2|feature3|features     |scaled_features |
+--------+--------+--------+-------------+----------------+
|1.0     |2.0     |3.0     |[1.0,2.0,3.0]|[-1.0,-1.0,-1.0]|
|2.0     |3.0     |4.0     |[2.0,3.0,4.0]|[0.0,0.0,0.0]   |
|3.0     |4.0     |5.0     |[3.0,4.0,5.0]|[1.0,1.0,1.0]   |
+--------+--------+--------+-------------+----------------+
AI 代码解读

将向量 Vector 拆分为列

从上面的示例中,可以看到 scaled_features 列就是规范化处理后的数据,但还需要进一步处理该数据,取到每列数据。

因为在进行数据预测时,我们使用的特征一般都是整型或浮点型,然而我们模型训练后的数据 scaled_features 列是一个向量类型的数据,注意,它不是一个数组!那么该如何取出该列中的每个元素呢?

在 Spark 3.0 以后提供了 vector_to_array() 方法用于将向量转换为数组,如下所示:

        val dataFrame: DataFrame = scaledData
                .withColumn("new_scaled_features", vector_to_array(col("scaled_features")))

        dataFrame.show(false)
AI 代码解读

其中 withColumn() 方法将添加一个新的名为 "new_scaled_features" 的列到新的 DataFrame 中,将 "scaled_features" 列向量转换成数组作为值。

输出结果如下所示:

+--------+--------+--------+-------------+----------------+-------------------+
|feature1|feature2|feature3|features     |scaled_features |new_scaled_features|
+--------+--------+--------+-------------+----------------+-------------------+
|1.0     |2.0     |3.0     |[1.0,2.0,3.0]|[-1.0,-1.0,-1.0]|[-1.0, -1.0, -1.0] |
|2.0     |3.0     |4.0     |[2.0,3.0,4.0]|[0.0,0.0,0.0]   |[0.0, 0.0, 0.0]    |
|3.0     |4.0     |5.0     |[3.0,4.0,5.0]|[1.0,1.0,1.0]   |[1.0, 1.0, 1.0]    |
+--------+--------+--------+-------------+----------------+-------------------+
AI 代码解读

其中 new_scaled_features 列就是我们转换成数组之后的列,现在我们就可以对其进行直接提取啦,如下所示:

        dataFrame
                .select(
                    col("new_scaled_features").getItem(0).alias("first"),
                    col("new_scaled_features").getItem(1).alias("second"),
                    col("new_scaled_features").getItem(2).alias("third")
                )
                .show(false)
AI 代码解读

输出结果如下:

+-----+------+-----+
|first|second|third|
+-----+------+-----+
|-1.0 |-1.0  |-1.0 |
|0.0  |0.0   |0.0  |
|1.0  |1.0   |1.0  |
+-----+------+-----+
AI 代码解读

向量拆分技巧 GOT!

目录
打赏
0
6
6
0
31
分享
相关文章
Pandas数据应用:机器学习预处理
本文介绍如何使用Pandas进行机器学习数据预处理,涵盖数据加载、缺失值处理、类型转换、标准化与归一化及分类变量编码等内容。常见问题包括文件路径错误、编码不正确、数据类型不符、缺失值处理不当等。通过代码案例详细解释每一步骤,并提供解决方案,确保数据质量,提升模型性能。
170 88
阿里云 EMR Serverless Spark 在微财机器学习场景下的应用
面对机器学习场景下的训练瓶颈,微财选择基于阿里云 EMR Serverless Spark 建立数据平台。通过 EMR Serverless Spark,微财突破了单机训练使用的数据规模瓶颈,大幅提升了训练效率,解决了存算分离架构下 Shuffle 稳定性和性能困扰,为智能风控等业务提供了强有力的技术支撑。
158 15
基于机器学习的数据分析:PLC采集的生产数据预测设备故障模型
本文介绍如何利用Python和Scikit-learn构建基于PLC数据的设备故障预测模型。通过实时采集温度、振动、电流等参数,进行数据预处理和特征提取,选择合适的机器学习模型(如随机森林、XGBoost),并优化模型性能。文章还分享了边缘计算部署方案及常见问题排查,强调模型预测应结合定期维护,确保系统稳定运行。
75 0
机器学习在生物信息学中的创新应用:解锁生物数据的奥秘
机器学习在生物信息学中的创新应用:解锁生物数据的奥秘
330 36
Diff-Instruct:指导任意生成模型训练的通用框架,无需额外训练数据即可提升生成质量
Diff-Instruct 是一种从预训练扩散模型中迁移知识的通用框架,通过最小化积分Kullback-Leibler散度,指导其他生成模型的训练,提升生成性能。
100 11
Diff-Instruct:指导任意生成模型训练的通用框架,无需额外训练数据即可提升生成质量
跨越鸿沟:PAI-DSW 支持动态数据挂载新体验
本文讲述了如何在 PAI-DSW 中集成和利用 Fluid 框架,以及通过动态挂载技术实现 OSS 等存储介质上数据集的快速接入和管理。通过案例演示,进一步展示了动态挂载功能的实际应用效果和优势。
模型训练数据-MinerU一款Pdf转Markdown软件
MinerU是由上海人工智能实验室OpenDataLab团队开发的开源智能数据提取工具,专长于复杂PDF文档的高效解析与提取。它能够将含有图片、公式、表格等多模态内容的PDF文档转化为Markdown格式,同时支持从网页和电子书中提取内容,显著提升了AI语料准备的效率。MinerU具备高精度的PDF模型解析工具链,能自动识别乱码,保留文档结构,并将公式转换为LaTeX格式,广泛适用于学术、财务、法律等领域。
436 4
谷歌提出视觉记忆方法,让大模型训练数据更灵活
谷歌研究人员提出了一种名为“视觉记忆”的方法,结合了深度神经网络的表示能力和数据库的灵活性。该方法将图像分类任务分为图像相似性和搜索两部分,支持灵活添加和删除数据、可解释的决策机制以及大规模数据处理能力。实验结果显示,该方法在多个数据集上取得了优异的性能,如在ImageNet上实现88.5%的top-1准确率。尽管有依赖预训练模型等限制,但视觉记忆为深度学习提供了新的思路。
73 2
揭秘机器学习背后的神秘力量:如何高效收集数据,让AI更懂你?
【10月更文挑战第12天】在数据驱动的时代,机器学习广泛应用,从智能推荐到自动驾驶。本文以电商平台个性化推荐系统为例,探讨数据收集方法,包括明确数据需求、选择数据来源、编写代码自动化收集、数据清洗与预处理及特征工程,最终完成数据的训练集和测试集划分,为模型训练奠定基础。
153 3
“探秘机器学习的幕后英雄:梯度下降——如何在数据的海洋中寻找那枚失落的钥匙?”
【10月更文挑战第11天】梯度下降是机器学习和深度学习中的核心优化算法,用于最小化损失函数,找到最优参数。通过计算损失函数的梯度,算法沿着负梯度方向更新参数,逐步逼近最小值。常见的变种包括批量梯度下降、随机梯度下降和小批量梯度下降,各有优缺点。示例代码展示了如何用Python和NumPy实现简单的线性回归模型训练。掌握梯度下降有助于深入理解模型优化机制。
67 2

热门文章

最新文章

AI助理

你好,我是AI助理

可以解答问题、推荐解决方案等