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