def train_model(self, net, train_dataloader, test_dataloader, device, logger, loss_train_list, acc_train_list, attn_train_list, loss_test_list, acc_test_list, attn_test_list): record_time(self.train_time_list, self._train_model, [ net, train_dataloader, test_dataloader, device, logger, loss_train_list, acc_train_list, attn_train_list, loss_test_list, acc_test_list, attn_test_list ])
def predict(self, net, dataloader, device): return record_time(self.test_time_list, self._predict, [ net, dataloader, device ])
def predict(self, clf, x): return record_time(self.test_time_list, self._predict, [clf, x])
def train_model(self, clf, x_train, y_train): record_time(self.train_time_list, self._train_model, [clf, x_train, y_train])