讲师中心 微信公众号
AI工具推荐 视频效率加速

如何使用Python在TensorFlow中处理结构化数据的特征工程?

酷伟小哥_3435

酷伟小哥_3435

发布时间:2026-06-17 07:32:00

|

260人浏览过

|

来源于php中文网

原创

TensorFlow中特征清洗需用tf.data.map内纯TF操作,而非Pandas;tf.feature_column仅声明格式不执行转换;数值列须预填充NaN,类别列需预清洗并匹配vocab size,Normalization层参数须固化保存。

如何使用python在tensorflow中处理结构化数据的特征工程?

TensorFlow 本身不提供 Pandas 那样的结构化数据清洗接口,直接用 tf.data 处理原始 CSV 或 DataFrame 容易踩坑——尤其是缺失值、类别编码、数值归一化混在一起时,模型训练会静默失败或收敛异常。

用 tf.feature_column 做特征定义,而不是预处理

很多人误以为 tf.feature_column 是“做特征工程”的工具,其实它只是把已有字段声明为模型可读的输入格式。它不执行实际转换(比如填缺失值、one-hot 编码),只在 tf.keras.layers.DenseFeatures 层中触发运行时转换。

  • tf.feature_column.categorical_column_with_vocabulary_list 要求输入已经是字符串或整数,不会自动把 None 或空字符串转成 UNK ——你得先用 Pandas 或 tf.strings.regex_replace 清洗
  • tf.feature_column.numeric_column 对 NaN 敏感:传入含 NaN 的张量会报 InvalidArgumentError: Nan in summary histogram,必须提前填充(如用中位数)
  • 所有 feature_column 的 default_value 参数只对 tf.parse_example 解析 TFRecord 时生效,对直接喂 pd.DataFrame 无效

把清洗逻辑写进 tf.data.Dataset.map,而非 Python 循环

用 Pandas 先清洗再转 tf.data.Dataset 看似简单,但无法和分布式训练、自动分片、prefetch 流水线对齐;而把清洗逻辑塞进 map 函数里,能保证每个样本在送入模型前都经过一致处理。

python-script-generator
python-script-generator

快速生成专业的 Python 脚本和应用代码。一键创建完整项目结构,支持CLI、API、爬虫、Bot、Django等多种项目类型,包含完整的项目结构、配置文件、依赖管理、测试、README和文档。

下载
  • 用 tf.py_function 包裹 Pandas 操作(例如 df.fillna())可行,但会退出图模式,损失性能且不能导出 SavedModel
  • 推荐组合使用 tf.strings.to_number、tf.where、tf.math.is_nan 实现纯 TensorFlow 清洗,例如:
    def clean_age(x):
      x = tf.strings.to_number(x, out_type=tf.float32)
      x = tf.where(tf.math.is_nan(x), tf.constant(35.0), x)
      return tf.clip_by_value(x, 0.0, 100.0)
  • 类别列做 tf.lookup.StaticHashTable 映射比 categorical_column_with_vocabulary_list 更可控,尤其当训练/推理 vocab 不一致时

数值特征标准化必须在训练集上 fit,且保存参数

tf.keras.layers.Normalization 默认是训练时动态统计均值/方差,但部署时若用新数据调用 adapt() 会覆盖原参数,导致线上预测偏移。

立即学习“Python免费学习笔记(深入)”;

  • 必须显式调用 layer.adapt(train_dataset.map(lambda x, y: x["age"])),并在保存模型前用 model.save(..., save_format="tf") 连带 layer 状态一起固化
  • 避免在 map 中用 np.mean() 或 tf.reduce_mean 实时计算——这会让每个 batch 的归一化基准不同,破坏训练稳定性
  • 如果用 Scikit-learn 的 StandardScaler 预处理,记得把 scaler.mean_ 和 scaler.scale_ 写进 tf.constant,再封装成自定义层,否则 SavedModel 加载后无法复现变换

最常被忽略的一点:tf.feature_column 的嵌入维度(dimension)和 tf.keras.layers.Embedding 的 input_dim 必须严格对应 vocab size + 1(含 default_value 占位符),否则 lookup 时索引越界,错误信息只显示 Invalid argument: indices[0] = 12345 is not in [0, 1000)——这个 1000 是你没意识到的隐式截断结果。

热门AI工具

更多
UP简历
UP简历 Hot

一款AI办公效率工具,主要用于基于AI技术的免费在线简历制作工具,适合需要提升相关任务效率的用户。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

PixTV
PixTV Hot

PixTV是一款面向AIGC内容创作的AI视频生成工具。

超级简历WonderCV

一款AI办公效率工具,主要用于免费求职简历模版下载制作,应届生职场人必备简历制作神器,适合需要提升相关任务效率的用户。

蛙蛙写作

一款AI论文写作工具,主要用于超级AI智能写作助手,适合需要提升相关任务效率的用户。

LibLibAI
LibLibAI Hot

一款AI视频创作工具,主要用于国内领先的AI创意平台,以海量模型、低门槛操作与“创作-分享-商业化”生态,让小白与专业创作者都能高效实现图文乃至视频创意表达,适合需要提升相关任务效率的用户。

PixPix
PixPix Hot

PixPix是一款面向电商视觉生产的AI商品图生成工具。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

相关专题

更多
python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

1691

2023.07.20

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

4264

2023.07.25

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

1689

2023.07.31

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

24857

2023.08.03

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

3027

2023.08.04

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

3067

2023.08.04

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

1163

2023.08.11

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

596

2023.08.10

Kratos框架HTTP与gRPC服务开发教程
Kratos框架HTTP与gRPC服务开发教程

本专题围绕Kratos框架双协议服务开发,涵盖HTTP路由与处理器编写、参数获取、gRPC服务实现与客户端调用、metadata上下文传递、encoding编解码注册、统一响应封装、超时控制与流式响应实现方法。

0

2026.10.10

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
TensorFlow 官方文档与教程
TensorFlow 官方文档与教程

共0课时 | 0人学习

Conan 2 安装指南
Conan 2 安装指南

共0课时 | 0人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn