Skip to content

set_node_label

Signature/Parameters

def set_node_label(self, nodes_label)

Update display labels for one or more nodes.

Parameters:

Name Type Description Default
nodes_label dict[str, str]

Mapping from node names to their new label strings.

 required

Examples:

>>> dag = DAG(graph="X -> Y")
>>> dag.set_node_label({"X": "Treatment (X)", "Y": "Outcome (Y)"})
Source code in causalinf/gcm.py 
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
def set_node_label(self, nodes_label):
    """
    Update display labels for one or more nodes.

    Parameters
    ----------
    nodes_label : dict[str, str]
        Mapping from node names to their new label strings.

    Examples
    --------
    >>> dag = DAG(graph="X -> Y")
    >>> dag.set_node_label({"X": "Treatment (X)", "Y": "Outcome (Y)"})
    """
    for node, label in nodes_label.items():
        self.nodes_label[node] = label