DPACMAN / configs /model /classifier.yaml
svincoff's picture
added dropout and overfit prevention
9da03b7
raw
history blame contribute delete
186 Bytes
_target_: dpacman.classifier.model.BindPredictor
lr: 1e-4
alpha: 20
gamma: 20
weight_decay: 0.01
dropout: 0.1
glm_input_dim: 1029
compressed_dim: 1029
hidden_dim: 256
loss_type: mixed