Stanford CS 224W (Winter 2021) Project
Authors: Sharan Ramjee ([email protected]) and Michael Wornow ([email protected])
- Clone the repository
- Install all the dependencies (WEGL depencies and imbalanced-learn)
- Download the data from here and extract it in the data folder
- Run the following command:
$ python models.py
The command uses the saved embeddings obtained using the Linear WEGL model, applied SMOTE on it, trains a HGBT model, and runs an OGB evaluator to evaluate the model