graphical_models.classes.mags.ancestral_graph.AncestralGraph.neighbors_of

AncestralGraph.neighbors_of(nodes: Union[Hashable, Set[Hashable]]) → Set[Hashable]

Return the neighbors of the node or set of nodes nodes.

Parameters

nodes – Nodes.

Examples

>>> from graphical_models import AncestralGraph
>>> g = AncestralGraph(directed={(1, 3), (2, 3)}, undirected={(1, 4), (2, 5)})
>>> g.neighbors_of(1)
{4}
>>> g.neighbors_of({1, 2})
{4, 5}