最新动态
【深度学习入门案例】波士顿房价预测
2025-01-02 11:15
 
 
 
 

数据下载

如果你还没安装wget,点击教程安装:window配置安装wget
下载数据

 
 

开始处理

 

模型设计

两步走
定义init函数:在类的初始化函数中声明每一层网络的实现函数。在房价预测模型中,只需要定义一层全连接层,模型结构和使用Python和Numpy构建神经网络模型》章节模型保持一致。
定义forward函数:构建神经网络结构,实现前向计算过程,并返回预测结果,在本任务中返回的是房价预测结果。

 

训练配置

1.声明定义好的回归模型Regressor实例,并将模型的状态设置为训练。
2.使用load_data函数加载训练数据和测试数据。
3.设置优化算法和学习率,优化算法采用随机梯度下降SGD,学习率设置为0.01。
代码为

 

注意
模型实例有两种状态:训练状态.train()和预测状态.eval()。训练时要执行正向计算和反向传播梯度两个过程,而预测时只需要执行正向计算,为模型指定运行状态

训练过程

 
 

保存模型

将模型当前的参数数据model.state_dict()保存到文件中(通过参数指定保存的文件名 LR_model,以备预测或校验的程序调用。
代码为

 

测试模型

通过load_one_example函数实现从数据集中抽一条样本作为测试样本,具体实现代码如下所示。

 
 

参考资料

百度深度学习飞桨

 
 

    以上就是本篇文章【【深度学习入门案例】波士顿房价预测】的全部内容了,欢迎阅览 ! 文章地址:https://sicmodule.kub2b.com/quote/17936.html 
     栏目首页      相关文章      动态      同类文章      热门文章      网站地图      返回首页 企库往资讯移动站https://sicmodule.kub2b.com/mobile/,查看更多   
发表评论
0评