17 lines
311 B
Python
17 lines
311 B
Python
# word2vec & scaler
|
|
EMBEDDING_SIZE = 100
|
|
|
|
# Cores to use while fitting word2vec vectors
|
|
WORD2VEC_WORKERS = 4
|
|
MIN_WORD_COUNT = 5
|
|
WORD2VEC_CONTEXT = 5
|
|
|
|
# Models
|
|
NN_ARCHITECTURE = 'cnn'
|
|
|
|
# Training parameters
|
|
BATCH_SIZE = 64
|
|
EPOCHS = 1
|
|
|
|
# Number of tokens to save from the abstract, zero padded
|
|
SAMPLE_LENGTH = 200 |