def test():
model.load_state_dict(torch.load(checkpt_file))
model.eval()
with torch.no_grad():
output = model(features[idx_test])
micro_test = muticlass_f1(output, labels[idx_test])
return micro_test.item()
def validate():
model.eval()
with torch.no_grad():
output = model(features[idx_val])
micro_val = muticlass_f1(output, labels[idx_val])
return micro_val.item()