R中neuralnet与caret训练的神经网络绘图神经元显示异常咨询

问题解答 #

1. 绘图操作正确性判断 #

你调用plot(nn_A[[4]]$finalModel)的操作完全正确，没有语法或调用逻辑问题。

2. 神经元数量显示异常的原因 #

该异常是neuralnet包自带可视化方法的已知兼容问题：
你第4个模型的最优结构中layer2=0，意味着第二隐藏层不存在，网络实际为「输入层→第一隐藏层（2个神经元）→第三隐藏层（5个神经元）→输出层」的跳层结构。而neuralnet的原生plot()方法没有针对中间隐藏层神经元数为0的场景做适配，因此会出现隐藏层计数、神经元数量展示错乱的问题。

3. 修复方案 #

可以任选以下两种方案解决：

• 调整超参数搜索逻辑：如果不需要支持跳层结构，可以把参数搜索网格中的中间隐藏层最小值改为1，避免出现空层，修改后的网格代码如下：

tune.grid.neuralnet <- expand.grid(
  layer1 = c(1:5),
  layer2 = c(1:5), # 最小值改为1，避免第二隐藏层为空
  layer3 = c(0:5)
)

• 更换可视化工具：使用兼容性更好的NeuralNetTools包的plotnet()函数绘图，该函数可正确识别空层、展示跳层结构，调用代码如下：

# 先安装包（如果未安装）
# install.packages("NeuralNetTools")
library(NeuralNetTools)
plotnet(nn_A[[4]]$finalModel)

内容的提问来源于stack exchange，提问作者antecessor