def _train_model(self, train_set, validation_set):
    '" Train the Model \n        \n        INPUT PARAMETER train set: Training set \n        INPUT PARAMETER validation_set: Validation set \n        OUTPUT self\n        '
    for i in range(self.num_epochs):
        self._run_train_step(train_set)
        if (validation_set is not None):
            self._run_validation_error(i, validation_set)
" Train the Model 

INPUT PARAMETER train set: Training set 
INPUT PARAMETER validation_set: Validation set 
OUTPUT self
def _train_model(self, train_set, validation_set):
    '" Train the Model \n        \n        INPUT PARAMETER train set: Training set \n        INPUT PARAMETER validation_set: Validation set \n        OUTPUT self\n        '
    for i in range(self.num_epochs):
        self._run_train_step(train_set)
        if (validation_set is not None):
            self._run_validation_error(i, validation_set)