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)[source]#

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

forward(g, inputs)[source]#

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