wandb.data_types.Graph

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
)