4332_assignment2

Files:

Our submission contains 5 main files: Graph_Analysis.ipynb, Initial_Tuning.ipynb, Best_Model.ipynb, pred.csv, README.md

  • Graphy.ipynb: exploring graph with some visualization
  • Initial_Tuning.ipynb: Our initial trials to fintune the parameters of deepwalk and node2vector
  • Best_Model.ipynb: Tuning hyperparameters of epochs and learning rate of word2vector, and find the best model to finish out prediction task.

Results:

Our best model performs 0.9391 in its AUC-ROC score.

Reminder

When checking our code, please follow the above instruction. The most important file which contains out final result is Best_Model.ipynb