qualia_codegen_core.graph.ModelGraph module

class qualia_codegen_core.graph.ModelGraph.ModelGraph(nodes: list[LayerNode] | None = None)[source]

Bases: object

add_node(node: LayerNode, innodes: Iterable[LayerNode] | None = None, outnodes: Iterable[LayerNode] | None = None) → None[source]

delete_node(node: LayerNode) → None[source]

delete_node_if(predicate: Callable[[LayerNode], bool]) → None[source]

replace_node(oldnode: LayerNode, newnode: LayerNode) → None[source]

find_node_from_layer(layer: TBaseLayer) → LayerNode | None[source]

get_nodes_for_layers(layers: TBaseLayer | Iterable[TBaseLayer]) → tuple[LayerNode | None, ...][source]

no_none_in_nodes(nodes: Iterable[LayerNode | None]) → TypeGuard[Iterable[LayerNode]][source]

add_layer(layer: TBaseLayer, inlayers: list[TBaseLayer] | None = None, outlayers: list[TBaseLayer] | None = None) → None[source]

property nodes: list[LayerNode]

graphviz() → str | None[source]

classmethod auto_detect(obj: keras.Model | nn.Module) → ModelGraph[source]