Skip to content

saithat/efficient-graphs

Repository files navigation

To start the training loop, run: cd floyd_net python dqn.py

Code directory: common: backend API functions to translate graphs between formats pytorch_structure2vec: pytorch implementation of S2V floyd_net: dqn.py: code for main training loop (i.e. get actions, update states, calculate loss, etc.) q_net.py: definition of DQN and full network pipeline message.py: random graph generation and efficiency calculation

About

Archive of final project for ECE517 and modification of "Improving the Robustness of Graphs through Reinforcement Learning and Graph Neural Networks"

Resources

Stars

1 star

Watchers

0 watching

Forks

Releases

No releases published

Packages

 
 
 

Contributors