Skip to content

Latest commit

 

History

History
19 lines (16 loc) · 777 Bytes

README.md

File metadata and controls

19 lines (16 loc) · 777 Bytes

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