-
Notifications
You must be signed in to change notification settings - Fork 0
/
constants.py
33 lines (29 loc) · 1.25 KB
/
constants.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
# Data files paths
TRAIN_DATA_NEGATIVE = './data/train_neg.txt'
TRAIN_DATA_POSITIVE = './data/train_pos.txt'
TRAIN_DATA_NEGATIVE_FULL = './data/train_neg_full.txt'
TRAIN_DATA_POSITIVE_FULL = './data/train_pos_full.txt'
TEST_DATA = './data/test_data.txt'
# Constants for BERT
BERT_WEIGHTS_PATH = './weights/bert/'
SUBMISSION_PATH_BERT = './submissions/bert/'
PREPROCESSED_DATA_PATH_BERT = './preprocessed_data/bert/'
PREPROCESSED_TRAIN_DATA_BERT = 'train_preprocessed.csv'
PREPROCESSED_TEST_DATA_BERT = 'test_preprocessed.csv'
# Constants for GRU
GLOVE_PATH = './data/glove.twitter.27B.100d.txt'
GRU_WEIGHTS_PATH = './weights/gru/'
SUBMISSION_PATH_GRU = './submissions/gru/'
PREPROCESSED_DATA_PATH_GRU = './preprocessed_data/gru/'
PREPROCESSED_TRAIN_DATA_GRU = 'train_preprocessed.csv'
PREPROCESSED_TEST_DATA_GRU = 'test_preprocessed.csv'
# Constants for baseline (classical ML)
CLASSICAL_WEIGHTS_PATH = './weights/baseline/'
SUBMISSION_PATH_CLASSICAL = './submissions/baseline/'
PREPROCESSED_DATA_PATH_CLASSICAL = './preprocessed_data/baseline/'
PREPROCESSED_TRAIN_DATA_CLASSICAL = 'train_preprocessed.csv'
PREPROCESSED_TEST_DATA_CLASSICAL = 'test_preprocessed.csv'
SEED = 666013
NJOBS = 15
# Enseble submission path
SUBMISSION_PATH_ENSEMBLE = './submissions/ensemble/'