Create symlink to dataset dir

category_classification/datasets/en/download_train_test.py

@@ -10,11 +10,12 @@ dataset = "hibiscus4000/arxiv-dataset"
 test_dataset = "arxiv_test.json"
 train_dataset = "arxiv_train.json"
 
+file_dir = Path(__file__).parent
 dataset_path = Path(dataset_download(dataset))
 test_file_path = dataset_path / test_dataset
 train_file_path = dataset_path / train_dataset
 
 if not test_file_path.exists():
-    os.symlink(dataset_path / test_dataset, test_dataset)
+    os.symlink(dataset_path / test_dataset, file_dir / test_dataset)
 if not train_file_path.exists():
-    os.symlink(dataset_path / train_dataset, train_dataset)
+    os.symlink(dataset_path / train_dataset, file_dir / train_dataset)

category_classification/datasets/ru/download_train_test.py

@@ -10,11 +10,12 @@ dataset = "hibiscus4000/Arxiv-papers-ru"
 test_dataset = "arxiv_test.csv"
 train_dataset = "arxiv_train.csv"
 
+file_dir = Path(__file__).parent
 dataset_path = Path(dataset_download(dataset))
 test_file_path = dataset_path / test_dataset
 train_file_path = dataset_path / train_dataset
 
 if not test_file_path.exists():
-    os.symlink(dataset_path / test_dataset, test_dataset)
+    os.symlink(dataset_path / test_dataset, file_dir / test_dataset)
 if not train_file_path.exists():
-    os.symlink(dataset_path / train_dataset, train_dataset)
+    os.symlink(dataset_path / train_dataset, file_dir / train_dataset)