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