saved_models/dimensions.json

{
  "drug_dim": 3042,
  "num_diseases": 1573,
  "num_drugs": 1410,
  "num_targets": 2018,
  "num_positive_samples": 42200,
  "train_test_split": 0.2
}