rl_equation_solver.agent.networks

Networks for agent policies

Classes

DQN(n_observations, n_actions, hidden_size)

Simple MLP network.

GCN(n_observations, n_actions, hidden_size)

Graph Convolution Network

LSTM(n_observations, n_actions, hidden_size, ...)

LSTM network