我正在尝试为乳腺癌数据集创建一个用于二元分类的神经网络:
https://www.kaggle.com/uciml/breast-cancer-wisconsin-data
我的神经网络由3层组成(不包括输入层):
-
第一层:6个神经元,使用tanh激活函数。
-
第二层:6个神经元,使用tanh激活函数。
-
最后一层:1个神经元,使用sigmoid激活函数。
遗憾的是,我的训练样本准确率只有约44%,测试样本准确率约为23%。
这是我的Python代码:
...
回答:
我认为你的反向传播有问题(我做了个快速测试,用Tensorflow尝试了你的模型,训练和测试数据的准确率都达到了约92%)。
我对你的代码做了以下修改:
...