diff --git a/dev-0/out.tsv b/dev-0/out.tsv index 9c158f8..6854ec7 100644 --- a/dev-0/out.tsv +++ b/dev-0/out.tsv @@ -1,3 +1,4 @@ +O O B-ORG O O O O O O O O O B-LOC O O B-MISC I-MISC O B-PER I-PER O O O O O O O B-ORG O B-ORG O O O O O O O O O O O O O O O O O O O O O O O O O O O O O O O O O O B-ORG O B-ORG I-ORG I-ORG O O O O O O B-ORG O O O O O O O O O O B-ORG O O O O B-ORG O O O O O O O O B-PER I-PER O B-ORG I-ORG O O O O O O O O O O O B-ORG O B-LOC O B-PER I-PER O O O O O O O O O O B-ORG O O O O O O O O O B-PER O O O O O O O O O O B-ORG O O O O O O O O O O O B-PER I-PER O B-PER I-PER O O O O B-ORG O O O O B-ORG O B-LOC O O B-PER O O O O B-LOC O O O O O I-PER O O O O O O O O O O B-ORG O O O O O O O O O O O O O O O B-ORG O O O O O O O O O O O O O O O O O O O O O O O O O O O O O O O O O O O O O O B-LOC O B-ORG O B-PER I-PER O O O O O B-LOC O O O O O O O O O O O O O O O O O O O O O O O O O B-ORG O O O O O O O B-LOC O B-PER I-PER O O O O B-ORG O O B-ORG O O O O O O B-ORG O O B-ORG O O O O O O O O O O O O O B-ORG O O O O O O O O O O O O O O O O O O O O B-MISC B-PER I-PER O O O O O B-PER I-PER O B-PER O O B-PER I-PER O O O O B-ORG O B-ORG O O O O O O B-ORG O O O O B-ORG I-ORG O O O O O O O O O O O O O B-ORG O O O O O O B-MISC O O O O O O O O O O B-ORG O O B-PER I-PER O O O O B-PER I-PER O O O O O O O O B-ORG O O O O O O O O O B-MISC I-MISC I-MISC O O O B-LOC O O O O O O O O O B-MISC O O O O O O O B-LOC O B-ORG O B-ORG O O O O O O O O B-ORG O O O O B-PER B-ORG I-ORG O O B-ORG O O O B-ORG O O O B-ORG O O O B-LOC O B-ORG O O O O B-PER I-PER O O B-PER I-PER O O B-PER O O O O B-ORG O O B-ORG I-PER O O O O O O B-PER I-PER O B-ORG O O B-PER I-PER O O B-PER O B-PER O O B-ORG O O O B-LOC O B-LOC I-LOC O O B-ORG O O B-ORG O O B-PER I-PER O O O O B-PER O O O B-PER I-PER O O B-PER I-PER O O B-PER I-PER O O O O B-LOC O B-ORG O O B-PER I-PER O O B-PER I-PER O O B-PER I-PER O O O B-ORG O O O O O O O B-ORG O B-ORG O O O O B-PER I-PER O O B-PER I-PER O O B-PER I-PER O O O B-ORG O O B-ORG O O B-ORG O B-ORG O O O O B-ORG I-ORG O B-PER B-PER I-PER O O O O O O B-PER I-PER O O B-PER I-PER O O O B-ORG O B-ORG O O O O B-PER I-PER O O O O O B-ORG O O B-PER O O O B-PER I-PER O O O B-PER O O O B-ORG I-ORG O O B-LOC O O B-LOC O O O O O O O O O O B-LOC O O O O O O O O O O O O O O O B-MISC I-MISC I-MISC O O O O O O B-LOC O O O O O O O O O O O O O O O O O O O O O O O O O O O O O B-MISC O O O O O B-MISC I-MISC O O O O O O O O O O O O O B-LOC O O B-MISC O O O O O O O B-ORG O B-LOC O O O O O O O O O O O B-ORG I-ORG I-ORG O O O O B-ORG O O O O O O O O O O B-ORG O O O O B-ORG O O O O O O O O B-LOC O O B-ORG O O O O O O O O O B-LOC I-LOC O O B-LOC O O O O O O O O O O O O B-LOC O O O O O B-ORG O B-ORG O O O O O O O O O O O O O O B-ORG O O O O O O O O B-ORG O O O O O B-LOC O B-LOC O O O B-MISC O O O O O O O O O O O O B-ORG O B-ORG O O O O O O O O O O O O O O O O O O B-MISC I-MISC O O B-ORG O O O O O O B-ORG O B-ORG O O O O O O O O O O O O O B-LOC I-LOC O B-LOC O O O O O B-ORG I-ORG O O O O O B-LOC O O O O B-ORG O O O O O O B-ORG O B-ORG O O O O O O O O O O O B-LOC O O O O O O O B-ORG O O O O O O O O O O O B-LOC I-LOC O B-LOC O O O O O B-ORG O O O O O O B-ORG I-ORG O O O B-LOC I-LOC O O O B-LOC I-LOC O O B-ORG O O B-LOC O O O B-LOC O O O O O O O B-PER I-PER O O O O O B-LOC O O O O O O O O O O B-ORG O O O O O O O O O O O O O O B-PER I-PER O O O O O O O B-MISC O O O O O O O B-PER I-PER O O B-PER O B-ORG O O O O O O B-LOC O O B-MISC I-MISC O O B-LOC O B-LOC O O B-PER O O O O O O O O O O O B-PER I-PER O O O O O O O O B-ORG O B-ORG O O O O O O O O O O O O O O O O O O O O O O O O B-PER O O O O O O O O I-PER O O O O O O O O O O O O O O O O O O O O B-PER O O O O O O O O O O O O O O O O O O O O O O O O O O O O O O O O O O O O O O O O O O B-ORG O O O O O O O O O O O O O O O O O O B-PER O B-MISC I-MISC O O O O O O O O O O O O O O O O O O O O O O O B-ORG O O B-PER I-PER B-ORG O diff --git a/main.py b/main.py index 4c9f123..529754a 100644 --- a/main.py +++ b/main.py @@ -76,12 +76,10 @@ mode = "eval" save_path = "dev-0/out.tsv" -data = pd.read_csv("dev-0/in.tsv", sep="\t") +data = pd.read_csv("dev-0/in.tsv", sep="\t", names=['0']) # data.columns = ["labels", "text"] -data.columns = ['0'] -ex_data = pd.read_csv("dev-0/expected.tsv", sep="\t") -ex_data.columns = ['labels'] +ex_data = pd.read_csv("dev-0/expected.tsv", sep="\t", names=['labels']) in_data = data["0"] target = ex_data["labels"]