polyaxon.libs.subgraph.SubGraph(mode, modules, name='Subgraph', features=None)
SubGraph is a class that represents the flow of layers.
str. Specifies if this training, evaluation or prediction. See
str. The name of this subgraph, used for creating the scope.
list. The modules to connect inside this subgraph, e.g. layers.
list. The list of features keys to extract and use in this subgraph. If
None, all features will be used.