/GNN_RL

reinforcement learning with pytorch geometric library

Primary LanguagePython

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