19 lines
397 B
Python
19 lines
397 B
Python
from torch import nn
|
|
|
|
|
|
class Network(nn.Module):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.hidden = nn.Linear(6, 70)
|
|
self.output = nn.Linear(70, 4)
|
|
|
|
self.sigmoid = nn.Sigmoid()
|
|
self.relu = nn.ReLU()
|
|
|
|
def forward(self, x):
|
|
x = self.hidden(x)
|
|
x = self.relu(x)
|
|
x = self.output(x)
|
|
x = self.sigmoid(x)
|
|
|
|
return x
|