神经网络欠拟合 – 乳腺癌数据集

我正在尝试为乳腺癌数据集创建一个用于二元分类的神经网络:

https://www.kaggle.com/uciml/breast-cancer-wisconsin-data

我的神经网络由3层组成(不包括输入层):

  • 第一层:6个神经元,使用tanh激活函数。

  • 第二层:6个神经元,使用tanh激活函数。

  • 最后一层:1个神经元,使用sigmoid激活函数。

遗憾的是,我的训练样本准确率只有约44%,测试样本准确率约为23%。

这是我的Python代码:

...

回答:

我认为你的反向传播有问题(我做了个快速测试,用Tensorflow尝试了你的模型,训练和测试数据的准确率都达到了约92%)。

我对你的代码做了以下修改:

...

Related Posts

多维度Top-k评分

例如,在机器学习中的自然语言处理中,通常使用波束搜索来…

R – Caret train() “错误:停止” 与 “在newdata中未找到对象中使用的所有变量名”

我正在尝试为蘑菇数据构建一个简单的朴素贝叶斯分类器。我…

创建训练和测试数据集分割,数据嵌套在多个文件夹中

我正在准备数据以训练一个图像识别模型。目前我有一个文件…

我的神经网络预测出现错误:IndexError: list index out of range

我正在进行一个简单的垃圾邮件/非垃圾邮件文本分类。我的…

python 给出数组是一维的,但索引了两个维度错误

我已经为 miniBatchGradientDesce…

TensorFlow自定义训练步骤使用不同的损失函数

背景 根据TensorFlow文档,可以使用以下方式执…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注