神经网络并不总是流行,部分原因是它们在某些情况下仍然计算成本高昂,部分原因是与支持向量机(SVM)等简单方法相比,它们似乎没有产生更好的结果。然而,最近神经网络变得流行起来。
在这篇文章中,我们将拟合神经网络,并将线性模型作为比较。
数据集
数据集是郊区房屋价格数据的集合。我们的目标是使用所有其他可用的连续变量来预测自住房屋(medv)的中位数。
首先,我们需要检查是否缺少数据点,否则我们需要填充数据集。
apply(data,2,function(x)sum(is.na(x)))
然后我们拟合线性回归模型并在测试集上进行测试。
index
MSE.lm
函数简单地从向量输出指定大小的随机选择样本的向量。
准备拟合神经网络
在拟合神经网络之前,需要做一些准备工作。神经网络不容易训练和调整。
作为_第一步_,我们将解决数据预处理问题。
因此,我们先划分数据:
maxs
scaled
train_
test_
请注意,需要转换为data.frame的矩阵。
参数
虽然有几个或多或少可接受的经验法则,但没有固定的规则可以使用多少层和神经元。一般一个隐藏层足以满足大量应用程序的需要。就神经元的数量而言,它应该在输入层大小和输出层大小之间,通常是输入大小的2/3
参数接受一个包含每个隐藏层的神经元数量的向量,而参数用于指定我们要进行回归或分类
绘制模型:
plot(nn)
这是模型的图形表示,每个连接都有权重:
黑色线条显示每个层与每个连接上的权重之间的连接,而蓝线显示每个步骤中添加的偏差项。偏差可以被认为是线性模型的截距。
视频LSTM神经网络架构和工作原理及其在Python中的预测应用
使用神经网络预测medv
现在我们可以尝试预测测试集的值并计算MSE。
pr.nn
然后我们比较两个MSE。
显然,在预测medv时,网络比线性模型做得更好。但是,这个结果取决于上面执行的训练测试集划分。下面,我们将进行快速交叉验证。
下面绘制了测试集上神经网络和线性模型性能的可视化结果
输出图:
通过检查图,我们可以看到神经网络的预测(通常)在直线周围更加集中(与线完美对齐将表明MSE为0,因此是理想的完美预测)。
下面绘制了模型比较:
交叉验证
交叉验证是构建预测模型的另一个非常重要的步骤。有不同类型的交叉验证方法。
然后通过计算平均误差,我们可以掌握模型。
我们将使用神经网络的for循环和线性模型的包中的函数来实现快速交叉验证。
据我所知,R中没有内置函数在这种神经网络上进行交叉验证。以下是线性模型的10折交叉验证MSE:
lm.fit
我以这种方式划分数据:90%的训练集和10%的测试集,随机方式进行10次。我使用库初始化进度条,因为神经网络的拟合可能需要一段时间。
过了一会儿,过程完成,我们计算平均MSE并将结果绘制成箱线图:
上面的代码输出以下boxplot:
神经网络的平均MSE(10.33)低于线性模型的MSE,尽管交叉验证的MSE似乎存在一定程度的变化。这可能取决于数据的划分或网络中权重的随机初始化。
领取专属 10元无门槛券
私享最新 技术干货