def __init__(self, A, num_features, num_layers, intermediate_size, num_instance=1, activation_fn=nn.ReLU, epochs=10, optimizer=optim.Adam, validation=False, **hyperparams):
    self.A = A
    self.num_features = num_features
    self.num_layers = num_layers
    self.activation_fn = activation_fn
    self.intermediate_size = intermediate_size
    self.epochs = epochs
    self.num_instance = num_instance
    self.validation = validation
    self.net = make_fc(num_layers=num_layers, num_features=num_features, activation_fn=activation_fn, intermediate_size=intermediate_size)
    self.optimizer = optimizer(self.net.parameters(), **hyperparams)
def __init__(self, A, num_features, num_layers, intermediate_size, num_instance=1, activation_fn=nn.ReLU, epochs=10, optimizer=optim.Adam, validation=False, **hyperparams):
    self.A = A
    self.num_features = num_features
    self.num_layers = num_layers
    self.activation_fn = activation_fn
    self.intermediate_size = intermediate_size
    self.epochs = epochs
    self.num_instance = num_instance
    self.validation = validation
    self.net = make_fc(num_layers=num_layers, num_features=num_features, activation_fn=activation_fn, intermediate_size=intermediate_size)
    self.optimizer = optimizer(self.net.parameters(), **hyperparams)