move files
This commit is contained in:
35
tutorial-contents/303_build_nn_quickly.py
Normal file
35
tutorial-contents/303_build_nn_quickly.py
Normal file
@ -0,0 +1,35 @@
|
||||
"""
|
||||
Know more, visit 莫烦Python: https://morvanzhou.github.io/tutorials/
|
||||
My Youtube Channel: https://www.youtube.com/user/MorvanZhou
|
||||
|
||||
Dependencies:
|
||||
torch: 0.1.11
|
||||
"""
|
||||
import torch
|
||||
import torch.nn.functional as F
|
||||
|
||||
|
||||
# replace following class code with an easy sequential network
|
||||
class Net(torch.nn.Module):
|
||||
def __init__(self, n_feature, n_hidden, n_output):
|
||||
super(Net, self).__init__()
|
||||
self.hidden = torch.nn.Linear(n_feature, n_hidden) # hidden layer
|
||||
self.predict = torch.nn.Linear(n_hidden, n_output) # output layer
|
||||
|
||||
def forward(self, x):
|
||||
x = F.relu(self.hidden(x)) # activation function for hidden layer
|
||||
x = self.predict(x) # linear output
|
||||
return x
|
||||
|
||||
net1 = Net(1, 10, 1)
|
||||
|
||||
# easy and fast way to build your network
|
||||
net2 = torch.nn.Sequential(
|
||||
torch.nn.Linear(1, 10),
|
||||
torch.nn.ReLU(),
|
||||
torch.nn.Linear(10, 1)
|
||||
)
|
||||
|
||||
|
||||
print(net1) # net1 architecture
|
||||
print(net2) # net2 architecture
|
||||
Reference in New Issue
Block a user