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])