|
1690 | 1690 | "metadata": {},
|
1691 | 1691 | "outputs": [],
|
1692 | 1692 | "source": [
|
1693 |
| - "dag5_7 = CausalGraphicalModel(\n", |
1694 |
| - " nodes=[\"M\", \"K\", \"N\"], edges=[(\"M\", \"K\"), (\"N\", \"K\"), (\"M\", \"N\")]\n", |
1695 |
| - ")\n", |
| 1693 | + "dag5_7 = nx.DiGraph()\n", |
| 1694 | + "dag5_7.add_edges_from([(\"M\", \"K\"), (\"N\", \"K\"), (\"M\", \"N\")])\n", |
1696 | 1695 | "coordinates = {\"M\": (0, 0.5), \"K\": (1, 1), \"N\": (2, 0.5)}\n",
|
1697 |
| - "nodes = list(dag5_7.dag.nodes.keys())\n", |
1698 |
| - "edges = list(dag5_7.dag.edges.keys())\n", |
1699 | 1696 | "MElist = []\n",
|
1700 | 1697 | "for i in range(2):\n",
|
1701 | 1698 | " for j in range(2):\n",
|
1702 | 1699 | " for k in range(2):\n",
|
1703 |
| - " try:\n", |
1704 |
| - " new_dag = CausalGraphicalModel(\n", |
1705 |
| - " nodes=nodes,\n", |
1706 |
| - " edges=[\n", |
1707 |
| - " edges[0] if i == 0 else edges[0][::-1],\n", |
1708 |
| - " edges[1] if j == 0 else edges[1][::-1],\n", |
1709 |
| - " edges[2] if k == 0 else edges[2][::-1],\n", |
1710 |
| - " ],\n", |
1711 |
| - " )\n", |
1712 |
| - " MElist.append(new_dag)\n", |
1713 |
| - " except:\n", |
1714 |
| - " pass" |
| 1700 | + " new_dag = nx.DiGraph()\n", |
| 1701 | + " new_dag.add_edges_from(\n", |
| 1702 | + " [edge[::-1] if flip else edge for edge, flip in zip(dag5_7.edges, (i, j, k))]\n", |
| 1703 | + " )\n", |
| 1704 | + " if not list(nx.simple_cycles(new_dag)):\n", |
| 1705 | + " MElist.append(new_dag)" |
1715 | 1706 | ]
|
1716 | 1707 | },
|
1717 | 1708 | {
|
|
0 commit comments