Update README.md

README.md

@@ -27,6 +27,17 @@ Binary classification. Given a drug SMILES string, predict the activity of BBB.
 ## Dataset statistics
 Total: 1,975; Train_val: 1,580; Test: 395
 
+## Pre-requisites
+Install the following packages
+```
+pip install PyTDC
+pip install DeepPurpose
+pip install git+https://github.com/bp-kelley/descriptastorus
+pip install dgl torch torchvision
+```
+You can also reference the colab notebook [here](https://colab.research.google.com/drive/1CL92SOCBS-eYDL99w8tjSNIG_ySXzMrG?usp=sharing)
+
+
 ## Dataset split
 Random split on 70% training, 10% validation, and 20% testing