是我大作业的代码了哈哈,要是需要带注释的可以私聊哈哈
代码使用方法:
1. 读取数据集
2. 初始化模型参数(选择哪个模型就初始化哪个)
3.定义激活函数(只有选择带隐含层的模型才需要激活函数)
4.防止过拟合(只有选择带隐含层的模型才需要防止过拟合,不过因为下面我们用的权重衰减在训练是也有定义,所以我们需要也给它加个定义)
5.定义模型(选择哪个模型就定义哪个)
6.定义损失函数
7. 定义优化函数
8.训练模型(训练结束可进行绘图)
9. 进行预测
可变参数:
1.选择模型(三种模型,可选用)
2.激活函数(两种)
3.防止过拟合(两种方法,可选用也可都用)
权重衰减(通过lambd 来调整, lambd = 0 即不使用此方法)
丢弃法(通过定义模型时选择)
4.损失函数(咱们的代码只用了一种)
5.优化方法(咱们的代码只用了一种)
6.训练模型(迭代周期数num_epochs和学习率lr可调)
7.定义数据集时小批量的大小