def _init_model(self, num_inputs):
    'Initializes internal linear model.\n\n        Args:\n            num_inputs: number of inputs that model will have.\n        '
    self.model = NNModel(num_inputs, self.num_hidden_layers, self.num_inner_features)
    self.optimizer = optim.Adam(self.model.parameters(), lr=self.learning_rate)
Initializes internal linear model.

Args:
    num_inputs: number of inputs that model will have.
def _init_model(self, num_inputs):
    'Initializes internal linear model.\n\n        Args:\n            num_inputs: number of inputs that model will have.\n        '
    self.model = NNModel(num_inputs, self.num_hidden_layers, self.num_inner_features)
    self.optimizer = optim.Adam(self.model.parameters(), lr=self.learning_rate)