Skip to content

reinforcement learning with pytorch geometric library

Notifications You must be signed in to change notification settings

keep9oing/GNN_RL

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 

Repository files navigation

GNN_RL

Intro

reinforcement learning with pytorch geometric library

Test feasibility of graph neural network on reinforcement learning framework.

REINFORCE algorithm with whitening baseline is used in Cartpole-v1 environment

GATConv(graph attention network) can also be the selection


Graph structure

construct graph with observation variables state


Action inferencing mechanism

process


Simulation result

reward_graph

About

reinforcement learning with pytorch geometric library

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages