人工智能学习Pytorch数据集分割及动量示例详解
作者：Swayzzu 发布时间：2021-04-29 11:28:55
标签：pytorch,人工智能,数据集,动量
1.数据集分割
通过datasets可以直接分别获取训练集和测试集。
通常我们会将训练集进行分割，通过torch.utils.data.random_split方法。
所有的数据都需要通过torch.util.data.DataLoader进行加载，才可以得到可以使用的数据集。
具体代码如下：
2.
2.正则化
PyTorch中的正则化和机器学习中的一样，不过设置方式不一样。
直接在优化器中，设置weight_decay即可。优化器中，默认的是L2范式，因此填入的参数就是lambda。想要使用L1范式的话，需要手动写出代码。
3.动量和学习率衰减
动量的设置可以直接在优化器中完成。通过momentum参数设置。
学习率的调整通过torch.optim.lr_scheduler中的ReduceLROnPlateau,StepLR实现。
ReduceLROnPlateau是自动检测损失值，并衰减学习率。
StepLR需要手动设置衰减的参数。
来源：https://blog.csdn.net/Swayzzu/article/details/121120412
0
投稿猜你喜欢
- 举一个例子，我现在有一些新闻信息，它包括这些字段；新闻ID，新闻Name，新闻ShortIntro，新闻Detail，新闻PublishTi
- python数值与字符串高级用法1.概述这篇是一篇没有尽头的文章，每当过段时间，再次打开就会看到不一样的内容，有新东西在更新啊。是啊，之所以
- 计算分页，嘿嘿一次搞定不用判断intNumPage = Abs(Int(-(intNumRecord/intPerPage)))
I. 前言在前面的一篇文章PyTorch搭建LSTM实现时间序列预测（负荷预测）中，我们利用LSTM实现了负荷预测，但我们只是简单利用负荷预- 目录安装sakila索引扫描排序表结构可以使用索引扫描来做排序的情况补足前导列order by 中只包含一种排序无法使用索引扫描的情况查询条
- MySQL目前不支持列的Default 为函数的形式,如达到你某列的默认值为当前更新日期与时间的功能,你可以使用TIMESTAMP列类型下面
- 请问如何用OleDbDataAdapter来对数据库进行删除、修改和添加？OleDbDataAdapter是DataSet和数据源之间建立联
- 在向大家详细介绍Linux mysql之前，首先让大家了解下Linux mysql，然后全面介绍Linux mysql，希望对大家有用。1.
- 有时候我们可能会把CSV中的数据导入到某个数据库的表中，比如做报表分析的时候。对于这个问题，我想一点也难不倒程序人员吧！但是要是SQL Se
- python包-pdfkit 将HTML转换为PDF什么是pdfkitpdfkit，把HTML+CSS格式的文件转换成PDF格式文档的一种工
- 在学习了一点 Python 基础之后，我们可以做一个罚点球的小游戏，大概流程是这样：每一轮，你先输入一个方向射门，然后电脑随机判断一个方向扑
- 问题：生产环境的数据库可能比较大，如果直接进行全备而不压缩的话，备份集就会占用了大量磁盘空间。给备份文件的存放管理带来不便。解决方案：通过w
- 概要：Oracle关系数据库系统以其卓越的性能获得了广泛的应用，而保证数据库的安全性是数据库管理工作的重要内容。本文是笔者在总结Oracle
- 踩了很多坑，记录一下这次试验，本次测试环境：Linux centos7 64位。pyenv是一个python版本管理工具，它能够进行全局的p
- Pandas count()与values_count()用法count()values_count()在指定的统计的列名上结果多了该列：对
- Windows 8 终于发布了，虽然现在可用的只是开发者预览版，好消息是，IE 10 也随着发了，虽然现在还只有Windows 8可用。我们
- 在python中，“np”一般是指“numpy”库，是第三方库“numpy”的别名。方法：利用命令“import numpy as np”将
- 引用计数在Python源码中，每一个对象都是一个结构体表示，都有一个计数字段。typedef struct_object { i
- 先推荐一个学习python的好网址简明 Python 教程 Swaroop, C. H. 著 沈洁元 译在线教程的网址
- 在使用pytorch的时候，经常会涉及到两种数据格式tensor和ndarray之间的转换，这里总结一下两种格式的转换：1. tensor
