From ca254cc58ada2577a4e3952bbcc9641819cf0e88 Mon Sep 17 00:00:00 2001 From: Claudio Scheer Date: Tue, 16 Jun 2020 19:43:42 -0300 Subject: [PATCH] Change dataset path --- src/seq2seq/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/seq2seq/main.py b/src/seq2seq/main.py index f5c24df..25f2703 100644 --- a/src/seq2seq/main.py +++ b/src/seq2seq/main.py @@ -9,7 +9,7 @@ transformers_logger = logging.getLogger("transformers") transformers_logger.setLevel(logging.WARNING) file_path = os.path.dirname(os.path.abspath(__file__)) -train_df = pd.read_csv(os.path.join(file_path, "../../dataset/input-target.csv")) +train_df = pd.read_csv(os.path.join(file_path, "../../dataset/input-target-256-dev.csv")) train_df.drop(train_df.columns[[0]], axis=1, inplace=True) train_df.dropna(subset=["input_text", "target_text"], inplace=True) -- GitLab