Skip to content

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

Notifications You must be signed in to change notification settings

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

Watchers

Forks

Releases

No releases published

Packages

No packages published