viz.renderers.ArchitectureGraph

export interface ArchitectureGraph {
  /** Graph nodes */
  nodes: ArchitectureNode[];
  /** Graph edges */
  edges: ArchitectureEdge[];
  /** Model name */
  name?: string;
  /** Total parameters */
  totalParams?: number;
}

nodes(ArchitectureNode[]): – Graph nodes
edges(ArchitectureEdge[]): – Graph edges
name(string)optional: – Model name
totalParams(number)optional: – Total parameters

Complete architecture graph.