|
| network |
| a dense model with one hidden layer
|
|
My dense neural network
Definition at line 23 of file simple.py.
◆ __init__()
__init__ |
( |
| self, |
|
|
| number_of_features ) |
Parameters:
number_of_features: number of input features
Definition at line 28 of file simple.py.
28 def __init__(self, number_of_features):
29 """
30 Parameters:
31 number_of_features: number of input features
32 """
33 super().__init__()
34
35
36 self.network = nn.Sequential(
37 nn.Linear(number_of_features, 128),
38 nn.ReLU(),
39 nn.Linear(128, 128),
40 nn.ReLU(),
41 nn.Linear(128, 1),
42 nn.Sigmoid(),
43 )
44
◆ forward()
Run the network
Definition at line 45 of file simple.py.
45 def forward(self, x):
46 """
47 Run the network
48 """
49 prob = self.network(x)
50 return prob
51
52
◆ network
Initial value:= nn.Sequential(
nn.Linear(number_of_features, 128),
nn.ReLU(),
nn.Linear(128, 128),
nn.ReLU(),
nn.Linear(128, 1),
nn.Sigmoid(),
)
a dense model with one hidden layer
Definition at line 36 of file simple.py.
The documentation for this class was generated from the following file: