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

TensorFlow二分类模型预测值全相同问题的根源与解决方案

大瑶大大_9369

大瑶大大_9369

发布时间:2026-07-04 15:52:06

|

833人浏览过

|

来源于php中文网

原创

本文详解tensorflow中binary crossentropy损失下模型输出恒定值(如全部为0.633)的根本原因——特征编码不当导致输入信息丢失,并提供从数据预处理、模型配置到训练监控的完整修复方案。

本文详解tensorflow中binary crossentropy损失下模型输出恒定值(如全部为0.633)的根本原因——特征编码不当导致输入信息丢失,并提供从数据预处理、模型配置到训练监控的完整修复方案。

在使用TensorFlow构建二分类模型时,若model.predict()返回所有样本都相同的概率值(例如全部为[[0.6335701], [0.6335701], ...]),这并非随机巧合,而是模型根本未学习到有效判别模式的明确信号。其核心原因往往隐藏在数据预处理环节——尤其是对原始特征的不当编码。

? 根本问题:pd.get_dummies() 的误用

在您的代码中:

x = pd.get_dummies(df.drop(['Last Result'], axis=1))

这一行是问题的关键所在。pd.get_dummies() 会将所有非数值列(包括可能存在的ID、时间戳、类别型字符串等)强制展开为稀疏的One-Hot编码。如果原始数据包含大量高基数类别特征(如用户ID、订单号)或本应保留序数/连续语义的字段(如“等级A/B/C”、“日期差值”),get_dummies 会:

  • 爆炸性增加特征维度(成百上千列),远超样本量;
  • 引入大量零值列,稀释有效信号;
  • 破坏原始特征的内在结构与统计分布,使模型无法捕捉任何有意义的模式;
  • 最终导致梯度消失或优化器停滞,模型退化为仅输出先验概率(≈正样本占比,即您观察到的 ~0.63)。

✅ 正确做法是:仅对真正需要编码的低基数类别特征进行One-Hot编码,其余数值型特征保持原样,高基数或特殊语义特征需单独处理。

TensorFlow Linux版 2.18.1
TensorFlow Linux版 2.18.1

TensorFlow 2.18.1 历史版本下载,来自 PyPI 官方发布,适合旧项目兼容、实验复现和指定环境安装。

下载

✅ 正确的数据预处理流程

import pandas as pd
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import StandardScaler, LabelEncoder
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Dense
from tensorflow.keras.optimizers import Adam  # 推荐替代SGD

# 1. 基础加载与目标编码
df = pd.read_csv('your_data.csv')
y = (df['Last Result'] == 'Registered').astype(int)  # 更清晰的布尔转整型

# 2. 特征分离(关键!)
x = df.drop(['Last Result'], axis=1).copy()

# 3. 智能特征处理(示例逻辑,请根据实际数据调整)
#   - 数值型列:直接保留
numeric_cols = x.select_dtypes(include=['number']).columns.tolist()
#   - 低基数类别列(如Status: ['Active','Inactive']):One-Hot编码
categorical_cols = x.select_dtypes(include=['object']).columns.tolist()
#     过滤高基数列(如ID列),避免爆炸
categorical_cols = [col for col in categorical_cols if x[col].nunique() < 10]
x_encoded = pd.get_dummies(x[numeric_cols + categorical_cols], 
                          columns=categorical_cols, drop_first=True)

# 4. 标准化(对数值特征至关重要!)
scaler = StandardScaler()
x_scaled = scaler.fit_transform(x_encoded)
x_train, x_test, y_train, y_test = train_test_split(
    x_scaled, y, test_size=0.2, stratify=y, random_state=42
)

? 模型与训练优化建议

除数据外,以下改进可显著提升收敛性:

# 使用更稳定的优化器和学习率
model = Sequential([
    Dense(64, activation='relu', input_shape=(x_train.shape[1],)),
    Dense(32, activation='relu'),
    Dense(1, activation='sigmoid')
])
model.compile(
    loss='binary_crossentropy',
    optimizer=Adam(learning_rate=0.001),  # 替代SGD
    metrics=['accuracy']
)

# 添加早停和学习率调度,防止过拟合/不收敛
from tensorflow.keras.callbacks import EarlyStopping, ReduceLROnPlateau

callbacks = [
    EarlyStopping(patience=15, restore_best_weights=True),
    ReduceLROnPlateau(factor=0.5, patience=5)
]

history = model.fit(
    x_train, y_train,
    epochs=200,
    batch_size=32,  # 避免过大batch_size(如128)导致梯度噪声
    validation_split=0.2,
    callbacks=callbacks,
    verbose=1
)

? 验证模型是否真正学习

训练后务必检查:

  • 损失曲线:训练/验证损失应持续下降且无剧烈震荡;
  • 预测分布:y_hat = model.predict(x_test) 应呈现合理分布(如0.1~0.9之间),而非单值;
  • 阈值评估:用y_pred = (y_hat > 0.5).astype(int)计算准确率、F1-score,而非直接用概率值。

⚠️ 注意:若数据本身存在严重类别不平衡(如95%负样本),即使模型正常训练,也可能因默认阈值0.5导致评估偏差。此时应使用classification_report或ROC-AUC,并考虑class_weight='balanced'参数。

通过修正特征工程逻辑、引入标准化、优化训练配置,模型将摆脱“恒定输出”的陷阱,真正学会从数据中提取判别性模式。记住:深度学习的成功,50%取决于数据质量,30%在于特征工程,仅20%属于模型架构本身。

热门AI工具

更多
豆包大模型

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

SkildArt
SkildArt Hot

SkildArt是一款AI文本写作工具,一站式 AI 视觉创作平台。

PixPix
PixPix Hot

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

WorkBuddy

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

切问学术

切问学术是一款AI论文写作工具,复旦大学NLP团队推出的AI学术智能体。

蛙蛙写作

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

DeepSeek

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

超级简历WonderCV

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

AionClaw
AionClaw Hot

AionClaw是一款面向办公、创作和编程任务的AI桌面智能体。

相关专题

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

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

1631

2023.07.20

python能做什么
python能做什么

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

3944

2023.07.25

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

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

1629

2023.07.31

python教程
python教程

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

22657

2023.08.03

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

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

2787

2023.08.04

python eval
python eval

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

2827

2023.08.04

scratch和python区别
scratch和python区别

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

1123

2023.08.11

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

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

596

2023.08.10

LLVM自定义Pass怎么写
LLVM自定义Pass怎么写

本专题聚焦LLVM自定义Pass开发,整理Pass类结构、run()方法、PreservedAnalyses、CMake构建、插件注册、-load-pass-plugin加载和测试用例编写流程。

0

2026.09.30

热门下载

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

精品课程

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

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