PatentSolver / LSTM /config.py
RedBaron5's picture
Duplicate from xin/PatentSolver
c13b805
raw
history blame contribute delete
No virus
515 Bytes
EMBEDDING_DIM = 50
MAX_SEQUENCE_LENGTH = 50
VALIDATION_SPLIT = 0.1
RATE_DROP_LSTM = 0.17
RATE_DROP_DENSE = 0.25
NUMBER_LSTM = 50
NUMBER_DENSE_UNITS = 50
ACTIVATION_FUNCTION = 'relu'
siamese_config = {
'EMBEDDING_DIM': EMBEDDING_DIM,
'MAX_SEQUENCE_LENGTH' : MAX_SEQUENCE_LENGTH,
'VALIDATION_SPLIT': VALIDATION_SPLIT,
'RATE_DROP_LSTM': RATE_DROP_LSTM,
'RATE_DROP_DENSE': RATE_DROP_DENSE,
'NUMBER_LSTM': NUMBER_LSTM,
'NUMBER_DENSE_UNITS': NUMBER_DENSE_UNITS,
'ACTIVATION_FUNCTION': ACTIVATION_FUNCTION
}