b站up主:刘二大人《PyTorch深度学习实践》
教程: https://www.bilibili.com/video/BV1Y7411d7Ys?p=6&vd_source=715b347a0d6cb8aa3822e5a102f366fe
两层模型
:
t
o
r
c
h
.
n
n
.
L
i
n
e
a
r
激活函数:
R
e
L
U
+
s
i
g
m
o
i
d
D
r
o
p
o
u
t
=
0.3
交叉熵损失函数:
n
n
.
B
C
E
L
o
s
s
优化器:
o
p
t
i
m
.
A
d
a
m
数据集:
h
t
t
p
s
:
/
/
w
w
w
.
k
a
g
g
l
e
.
c
o
m
/
c
o
m
p
e
t
i
t
i
o
n
s
/
t
i
t
a
n
i
c
两层模型:torch.nn.Linear \\激活函数:ReLU+sigmoid \\Dropout=0.3 \\交叉熵损失函数:nn.BCELoss \\优化器:optim.Adam \\数据集:https://www.kaggle.com/competitions/titanic
两层模型:torch.nn.Linear激活函数:ReLU+sigmoidDropout=0.3交叉熵损失函数:nn.BCELoss优化器:optim.Adam数据集:https://www.kaggle.com/competitions/titanic
模型结构:
两层全连接
+ dropout = 0.3
+ ReLU
+ sigmoid
训练过程:
Submission效果:
源码:https://www.kaggle.com/code/invokeg/titanic-pytorch-linear