write_dataset(test_dataset, "./data/test.json") def load_dataset_example(method=1): """ An example for loading dataset. :param method: 1 (by fold including tsv files) or 2 (by json) :return: gold_dataset, test_dataset """ # method 1: load by file fold if method == 1: gold_dataset = load_dataset(path="./data/golden", load_type=LoadTypeEnum.fold) test_dataset = load_dataset(path="./data/test", load_type=LoadTypeEnum.fold) else: gold_dataset = load_dataset(path="./data/golden.json", load_type=LoadTypeEnum.json) test_dataset = load_dataset(path="./data/test.json", load_type=LoadTypeEnum.json) return gold_dataset, test_dataset if __name__ == '__main__': for i in range(1, 3): print("方法" + str(i)) gold_dataset, test_dataset = load_dataset_example(method=i) evaluate = Evaluate() evaluate.calculate_results(test_dataset, gold_dataset) print(evaluate.to_string())