Please tell me, did I get it right here?
..for some reason, the results even got worse.
import torch.nn as nn
import torch.optim as optim
class MultipleRegression(nn.Module):
def init(self, num_features):
super(MultipleRegression, self).init()
self.layer_1 = nn.Linear(num_features, 16)
self.layer_2 = nn.Linear(16, 32)
self.layer_3 = nn.Linear(32, num_features)
self.layer_out = nn.Linear(num_features, 1)
self.relu = nn.ELU()
self.revin_layer = RevIN(num_features) # <<<-----
def forward(self, inputs):
x_in = self.revin_layer(inputs, 'norm')
x = self.relu(self.layer_1(x_in))
x = self.relu(self.layer_2(x))
x = self.relu(self.layer_3(x))
x = self.revin_layer(x, 'denorm')
x_out = self.layer_out(x)
return x_out
Please tell me, did I get it right here?
..for some reason, the results even got worse.
import torch.nn as nn
import torch.optim as optim
class MultipleRegression(nn.Module):
def init(self, num_features):
super(MultipleRegression, self).init()