From 180a662e07511151627391d568ca4eec466d8cb8 Mon Sep 17 00:00:00 2001 From: wangobango Date: Mon, 21 Jun 2021 18:08:57 +0200 Subject: [PATCH] fixed incorrect tsv read --- dev-0/out.tsv | 1 + main.py | 6 ++---- 2 files changed, 3 insertions(+), 4 deletions(-) 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"]