​​View source on GitHub​

Wandb class for graphs

Graph(
    format="keras"
)

This class is typically used for saving and diplaying neural net models. It represents the graph as an array of nodes and edges. The nodes can have labels that can be visualized by wandb.

Examples:

Import a keras model:

    Graph.from_keras(keras_model)

Methods

add_edge

​View source​

add_edge(
    from_node, to_node
)

add_node

​View source​

add_node(
    node=None, **node_kwargs
)

from_keras

​View source​

@classmethod
from_keras(
    model
)

pprint

​View source​

pprint()

__getitem__

​View source​

__getitem__(
    nid
)