rofunc.learning.RofuncRL.state_encoders.graph_encoders

Contents

`rofunc.learning.RofuncRL.state_encoders.graph_encoders`#

1. Module Contents#

1.1. Classes#

`HomoGraphEncoder`
`HeteroGraphEncoder`

1.2. API#

class rofunc.learning.RofuncRL.state_encoders.graph_encoders.HomoGraphEncoder(in_dim, hidden_dim)#

Bases: rofunc.learning.RofuncRL.state_encoders.base_encoders.BaseEncoder

forward(g, inputs)#

class rofunc.learning.RofuncRL.state_encoders.graph_encoders.HeteroGraphEncoder#: Bases: rofunc.learning.RofuncRL.state_encoders.base_encoders.BaseEncoder