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

Action inferencing mechanism

Simulation result