Documentation
Search…
wandb.data_types.Graph
Wandb class for graphs
1
Graph(
2
format="keras"
3
)
Copied!
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:
1
Graph.from_keras(keras_model)
Copied!

Methods

add_edge

1
add_edge(
2
from_node, to_node
3
)
Copied!

add_node

1
add_node(
2
node=None, **node_kwargs
3
)
Copied!

from_keras

1
@classmethod
2
from_keras(
3
model
4
)
Copied!

pprint

1
pprint()
Copied!

__getitem__

1
__getitem__(
2
nid
3
)
Copied!