怎么用python写一行dropout代码
时间: 2024-04-11 08:06:17 浏览: 77
机器学习——Dropout,python实现
4星 · 用户满意度95%
使用PyTorch框架,在搭建深度神经网络的时候可以添加dropout层来进行训练过程中的随机失活(dropout)操作,以防止过拟合。下面是一行使用PyTorch实现dropout的代码示例:
```python
import torch.nn as nn
model = nn.Sequential(
nn.Linear(100, 50),
nn.Dropout(0.5), # 50%的概率随机失活
nn.ReLU(),
nn.Linear(50, 10),
)
```
阅读全文