头歌-神经网络
【代码】头歌-神经网络。
·
第1关:网络层
import torch.nn as nn
myNet = nn.Sequential(
########## Begin ##########
nn.Linear(3, 100),
nn.Linear(100, 1000),
nn.Linear(1000, 100),
nn.Linear(100, 10)
)
########## End ##########
print(myNet)
第2关:激活函数
import torch
from torch.autograd import Variable
import torch.nn.functional as F
x = torch.linspace(-5, 5, 200) # 构造一段连续的数据
x = Variable(x)
########## Begin ##########
# 使用sigmoid激活函数
y_sigmoid = torch.sigmoid(x)
print("y_sigmoid:")
print(y_sigmoid)
# 使用tanh激活函数
y_tanh = torch.tanh(x)
print("y_tanh:")
print(y_tanh)
# 使用ReLU激活函数
y_relu = torch.relu(x)
print("y_relu:")
print(y_relu)
# 使用softplus激活函数
y_softplus = F.softplus(x)
print("y_softplus:")
print(y_softplus)
# 使用leakyReLU激活函数
y_leakyrelu = F.leaky_relu(x)
print("y_leakyrelu:")
print(y_leakyrelu)
# 使用ELU激活函数
y_ELU = F.elu(x)
print("y_ELU:")
print(y_ELU)
# 使用SELU激活函数
y_SELU = F.selu(x)
print("y_SELU:")
print(y_SELU)
########## End ##########
第3关:模型优化

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐



所有评论(0)