news 2026/6/9 21:17:15

Flink ML StandardScaler 标准化(去均值 + 除以标准差)让特征“同量纲”更好学

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
Flink ML StandardScaler 标准化(去均值 + 除以标准差)让特征“同量纲”更好学

1. StandardScaler 做什么?

对向量特征的每个维度 (x) 做标准化:

  • 先减去均值:(x - \mu)(可选)
  • 再除以标准差:((x - \mu) / \sigma)(可选)

在 Flink ML 里,通过两个开关控制:

  • withMean:是否减均值(默认 false)
  • withStd:是否除以标准差(默认 true)

2. 输入列与输出列

输入列(Input Columns)

参数名类型默认值说明
inputColVector"input"待标准化的特征向量

输出列(Output Columns)

参数名类型默认值说明
outputColVector"output"标准化后的向量

3. 参数详解(Parameters)

Key默认值类型说明
inputCol"input"String输入列名
outputCol"output"String输出列名
withMeanfalseBoolean是否先减去均值(中心化)
withStdtrueBoolean是否按标准差缩放到单位方差

withMean 什么时候开?

  • 你希望特征以 0 为中心、并且数据不是稀疏 one-hot/高维稀疏向量:可以开
  • 如果你的特征是稀疏向量(例如 OneHotEncoder 输出),一般不建议开(中心化会破坏稀疏性、带来不必要开销)

4. Java 示例解读(fit + transform)

标准用法永远是两步:

1)在训练数据上fit()学到每个维度的统计量(均值、方差/标准差)
2)用同一个StandardScalerModel对训练/预测数据transform(),保证线上线下一致

你给的示例在同一份 inputTable 上 fit + transform,演示效果更直观。

importorg.apache.flink.ml.feature.standardscaler.StandardScaler;importorg.apache.flink.ml.feature.standardscaler.StandardScalerModel;importorg.apache.flink.ml.linalg.DenseVector;importorg.apache.flink.ml.linalg.Vectors;importorg.apache.flink.streaming.api.datastream.DataStream;importorg.apache.flink.streaming.api.environment.StreamExecutionEnvironment;importorg.apache.flink.table.api.Table;importorg.apache.flink.table.api.bridge.java.StreamTableEnvironment;importorg.apache.flink.types.Row;importorg.apache.flink.util.CloseableIterator;publicclassStandardScalerExample{publicstaticvoidmain(String[]args){StreamExecutionEnvironmentenv=StreamExecutionEnvironment.getExecutionEnvironment();StreamTableEnvironmenttEnv=StreamTableEnvironment.create(env);DataStream<Row>inputStream=env.fromElements(Row.of(Vectors.dense(-2.5,9,1)),Row.of(Vectors.dense(1.4,-5,1)),Row.of(Vectors.dense(2,-1,-2)));TableinputTable=tEnv.fromDataStream(inputStream).as("input");StandardScalerstandardScaler=newStandardScaler();// 默认 withMean=false, withStd=trueStandardScalerModelmodel=standardScaler.fit(inputTable);TableoutputTable=model.transform(inputTable)[0];for(CloseableIterator<Row>it=outputTable.execute().collect();it.hasNext();){Rowrow=it.next();DenseVectorinputValue=(DenseVector)row.getField(standardScaler.getInputCol());DenseVectoroutputValue=(DenseVector)row.getField(standardScaler.getOutputCol());System.out.printf("Input Value: %s\tOutput Value: %s\n",inputValue,outputValue);}}}

这段代码做了什么:

  • fit():扫描 inputTable,计算每个维度的标准差(以及如果 withMean=true 则计算均值)
  • transform():对每条向量逐维标准化,新增输出列output

5. 实战建议

1)StandardScaler 比 MinMaxScaler 更抗异常值一点,但也不是“免疫”

StandardScaler用均值/方差,极端值仍会影响统计量,只是通常比 min/max 更稳定。
如果异常值特别多,建议先做截断/清洗,再标准化。

2)不要在预测数据上重新 fit

必须做到:

  • 训练阶段:fit(train)
  • 预测阶段:transform(predict) 用同一个 model
    否则线上每批数据的缩放尺度都变,模型输出会漂。

3)常见组合

  • VectorAssembler → StandardScaler → KMeans/KNN/LinearSVC/LogisticRegression
  • Bucketizer/OneHotEncoder 这种离散稀疏特征链路,一般不需要 withMean
版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/6/10 12:12:00

PyTorch镜像运行AutoML任务:自动化超参搜索实战

PyTorch镜像运行AutoML任务&#xff1a;自动化超参搜索实战 在深度学习模型研发中&#xff0c;一个常见的困境是&#xff1a;明明架构设计合理、数据质量也过关&#xff0c;但模型表现始终差那么一口气——问题往往出在超参数上。学习率设高了震荡不收敛&#xff0c;设低了训练…

作者头像 李华
网站建设 2026/6/10 12:13:51

PyTorch镜像中如何安装nvtop监控GPU温度?

PyTorch镜像中如何安装nvtop监控GPU温度&#xff1f; 在深度学习训练日益依赖多GPU并行的今天&#xff0c;一个看似不起眼却至关重要的问题逐渐浮现&#xff1a;我们真的了解GPU的实时状态吗&#xff1f; 很多开发者都经历过这样的场景——模型刚开始训练时速度飞快&#xff0c…

作者头像 李华
网站建设 2026/6/10 14:10:03

PyTorch镜像能否离线使用?打包导出与导入方法

PyTorch镜像能否离线使用&#xff1f;打包导出与导入方法 在现代AI项目部署中&#xff0c;一个常见的挑战是&#xff1a;如何在没有互联网连接的服务器上快速搭建一套稳定、可复用的深度学习环境&#xff1f;尤其是在企业内网、边缘设备或安全隔离网络中&#xff0c;传统依赖在…

作者头像 李华
网站建设 2026/6/10 16:54:24

PyTorch镜像中运行异构计算任务:CPU+GPU协同

PyTorch镜像中运行异构计算任务&#xff1a;CPUGPU协同 在深度学习项目开发过程中&#xff0c;你是否曾遇到过这样的场景&#xff1f;团队成员拿着同一份代码却因“在我机器上能跑”而陷入争执&#xff1b;新同事花三天才配好环境&#xff0c;结果训练时又报出CUDA版本不兼容的…

作者头像 李华