Neural-Nework Model

Model Definition

  • Input: Two drug fingerprints, each of size 1024.
  • Encoder: Two-layer fully connected neural network with ReLU activation.
  • Classifier: Merges encoded fingerprints and predicts interaction severity class (3 classes).
  • Data-split: Stratified split on 80% - train, 10% - validation, 10% - test
  • Training Epochs: 100 unless reached convergence in validation accuracy
  • Learning Rate: 0.001
  • Batch Size: 2048