{
  "cells": [
    {
      "cell_type": "markdown",
      "id": "frontmatter",
      "metadata": {},
      "source": [
        "---\n",
        "title: \"非マルコフ型誤り検査を用いた事後選択\"\n",
        "description: \"Qiskit Mitigationの最新バージョンにおける、非マルコフ型エラーチェックを用いたポストセレクション\"\n",
        "---\n",
        "\n"
      ]
    },
    {
      "cell_type": "markdown",
      "id": "69a159c7-51fd-4d6e-98ee-0458927295aa",
      "metadata": {},
      "source": [
        "<span id=\"postselection-with-non-markovian-error-checks\" />\n",
        "\n",
        "# 非マルコフ型誤り検査を用いた事後選択\n",
        "\n",
        "このガイドでは、非マルコフ型エラーチェックとは何か、およびソフトウェアでそれを実装する方法について説明します `qiskit-mitigation`。 まずは、2量子ビットの簡単な例から始め、検証手順を具体的にイメージし、その仕組みについて解説します。 直感をつかんだら、チェックを実装し、さまざまなポストセレクション戦略を比較する20キュービットの例題に取り組みます。\n",
        "\n",
        "<span id=\"what-are-non-markovian-error-checks\" />\n",
        "\n",
        "#### 非マルコフ型エラーチェックとは何ですか？\n",
        "\n",
        "非マルコフ型エラーチェックとは、量子ビットの状態が非マルコフ型エラー源によって破損している可能性が高いかどうかを、測定に基づいて検証する手法である。 このテストは、回路の冒頭で各量子ビットに持続時間が延長されたビット反転ゲート（`xslow` または、角度が $\\frac{\\pi}{N}$ である一連の $N$ ゲート `rx` ）を適用し、続いて通常の測定を行い `XGate` 、状態が $|0\\rangle \\mapsto |1\\rangle \\mapsto |0\\rangle$ から正常に反転したことを確認することで行われます。また、このテストは回路の終了時、各終端測定の後に実行することもでき、デバイスが量子ビットの測定済み状態を $|x\\rangle \\mapsto |x\\oplus1\\rangle$ から正常に反転できることを確認できます。 チェックに失敗したサンプルは信頼できないものとみなすことができ、すべてのチェックに合格したサンプルのみを選別することで、ノイズの多い分布のフィデリティを向上させることができる。\n",
        "\n",
        "<span id=\"strategies-for-postselecting-based-on-non-markovian-error-checks\" />\n",
        "\n",
        "#### 非マルコフ型誤りチェックに基づくポストセレクションの戦略\n",
        "\n",
        "* **ノードベースのポストセレクション** ：各量子ビットがチェックに合格したサンプルのみをポストセレクションの対象とする。 これは最も厳格な戦略であり、一般的により多くのサンプルが除外される結果となります。\n",
        "* **エッジに基づくポストセレクション** ：量子ビットの結合グラフにおける各エッジについて、少なくとも1つの量子ビットがチェックをクリアするようにする。 この戦略はより寛容であり、ある量子ビットがチェックに失敗した場合でも、その量子ビットの隣接するすべての量子ビットがチェックに合格していれば、そのサンプルは保持されることがある。\n",
        "\n",
        "<span id=\"spectator-qubits\" />\n",
        "\n",
        "#### スペクテーター量子ビット\n",
        "\n",
        "元の回路の実行に使用される量子ビットに隣接する「傍観者」アンシラ量子ビットを指定することができます。 これらのスペクテーターには、非マルコフ型エラーチェックを適用することができ、それによって、これらのチェックが検出するように設計されている非マルコフ型ノイズに対する検出範囲を広げることができる。\n",
        "\n"
      ]
    },
    {
      "cell_type": "markdown",
      "id": "5a7ce524-da69-41e2-948c-d2743ae9b603",
      "metadata": {},
      "source": [
        "<span id=\"part-1-visualize-non-markovian-error-checks\" />\n",
        "\n",
        "### パート1. 非マルコフ型のエラーチェックを可視化する\n",
        "\n",
        "QPU上でこの手法を実演する前に、まず2クビットの小さな回路を作成し、その動作を可視化するためのチェック機能を追加します。 まず、バック `ibm_kingston` エンドのQPUにトランスパイルされたベル状態から始めます。\n",
        "\n"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 1,
      "id": "1b52a0f7-7f16-4df3-94b9-06072db135b4",
      "metadata": {},
      "outputs": [
        {
          "data": {
            "text/plain": [
              "<Image src=\"/docs/images/addons/qiskit-mitigation/guides/postselection-with-non-markovian-error-checks/extracted-outputs/1b52a0f7-7f16-4df3-94b9-06072db135b4-0.avif\" alt=\"Output of the previous code cell\" />"
            ]
          },
          "execution_count": 1,
          "metadata": {},
          "output_type": "execute_result"
        }
      ],
      "source": [
        "from qiskit import QuantumCircuit\n",
        "from qiskit.transpiler import generate_preset_pass_manager\n",
        "from qiskit_ibm_runtime import QiskitRuntimeService\n",
        "\n",
        "service = QiskitRuntimeService(channel=\"ibm_quantum_platform\")\n",
        "backend = service.backend(\"ibm_kingston\")\n",
        "\n",
        "qc = QuantumCircuit(2)\n",
        "qc.h(0)\n",
        "qc.cx(0, 1)\n",
        "qc.measure_all()\n",
        "qc_isa = generate_preset_pass_manager(\n",
        "    optimization_level=0, backend=backend\n",
        ").run(qc)\n",
        "qc_isa.draw(\"mpl\")"
      ]
    },
    {
      "cell_type": "markdown",
      "id": "3336a9e9-01d2-4b57-afcf-2101edc19f26",
      "metadata": {},
      "source": [
        "回路の先頭に、非マルコフ型の誤りチェックをいくつか追加してください。 **チェックは、測定で終了するキュービットに対してのみ行われます**。 これらのチェックにより、デバイスが量子ビットの初期状態を $|0\\rangle \\mapsto |1\\rangle \\mapsto |0\\rangle$ から正常に反転させることができることが保証され、これらのチェックによる測定結果は、と呼ばれる新しい古典レジスタに読み出されます `meas_pre`。\n",
        "\n"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 2,
      "id": "5e75209c-74c9-44cf-80ea-1b0cbe9e3606",
      "metadata": {},
      "outputs": [
        {
          "data": {
            "text/plain": [
              "<Image src=\"/docs/images/addons/qiskit-mitigation/guides/postselection-with-non-markovian-error-checks/extracted-outputs/5e75209c-74c9-44cf-80ea-1b0cbe9e3606-0.avif\" alt=\"Output of the previous code cell\" />"
            ]
          },
          "execution_count": 2,
          "metadata": {},
          "output_type": "execute_result"
        }
      ],
      "source": [
        "from qiskit.transpiler import PassManager\n",
        "from qiskit_mitigation.postselection.passes import (\n",
        "    AddPreCircuitNonMarkovianErrorChecks,\n",
        ")\n",
        "\n",
        "qc_w_checks = PassManager([AddPreCircuitNonMarkovianErrorChecks()]).run(\n",
        "    qc_isa\n",
        ")\n",
        "qc_w_checks.draw(\"mpl\", idle_wires=False)"
      ]
    },
    {
      "cell_type": "markdown",
      "id": "7c9da765-6915-41c1-b869-e43deb71e0fc",
      "metadata": {},
      "source": [
        "同様に、各端子の測定後に、回路の末端にチェックを追加することもできます。 これらのチェックにより、デバイスが量子ビットの状態を $|x\\rangle \\mapsto |x\\oplus1\\rangle$ に切り替えることができることが保証され、その測定結果は、と呼ばれる新しい古典レジスタに読み出されます `meas_ps`。\n",
        "\n"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 3,
      "id": "2ed2b6e8-1c9d-4b10-9374-6908d5ba6b1f",
      "metadata": {},
      "outputs": [
        {
          "data": {
            "text/plain": [
              "<Image src=\"/docs/images/addons/qiskit-mitigation/guides/postselection-with-non-markovian-error-checks/extracted-outputs/2ed2b6e8-1c9d-4b10-9374-6908d5ba6b1f-0.avif\" alt=\"Output of the previous code cell\" />"
            ]
          },
          "execution_count": 3,
          "metadata": {},
          "output_type": "execute_result"
        }
      ],
      "source": [
        "from qiskit_mitigation.postselection.passes import (\n",
        "    AddPostCircuitNonMarkovianErrorChecks,\n",
        ")\n",
        "\n",
        "qc_w_checks = PassManager([AddPostCircuitNonMarkovianErrorChecks()]).run(\n",
        "    qc_w_checks\n",
        ")\n",
        "qc_w_checks.draw(\"mpl\", idle_wires=False)"
      ]
    },
    {
      "cell_type": "markdown",
      "id": "4b1a618a-66ef-4eb9-b062-e91fa51c81db",
      "metadata": {},
      "source": [
        "はじめに述べたように、データ量子ビットに隣接する補助量子ビットを「オブザーバー」として利用することで、エラーのカバー範囲を広げることができます。 サーキット走行前および走行後の観客確認を追加する。 物理量子ビット `2` がデータ量子ビット に隣接 `ibm_kingston` しているため `1`、それがオブザーバーとして追加されており、回路の前後でチェックが追加されていることがわかります。 観測値は、および `spec_pre` と呼ばれる新しい古典レジスタに追加されました `spec`。\n",
        "\n",
        "このような回路には、考えられるすべての非マルコフ型エラーチェックが含まれているため、この回路からサンプリングを行うことで、ユーザーは- \\[pre/post]回路チェックと- \\[node/edge]に基づくポストセレクション戦略の任意の組み合わせに基づいてポストセレクションを行うことが可能になる。 量子ハードウェア上での動作については、本ガイドの第2部をご覧ください。\n",
        "\n"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 4,
      "id": "29965ec1-0fea-4162-9fbf-8af972d8c76c",
      "metadata": {},
      "outputs": [
        {
          "data": {
            "text/plain": [
              "<Image src=\"/docs/images/addons/qiskit-mitigation/guides/postselection-with-non-markovian-error-checks/extracted-outputs/29965ec1-0fea-4162-9fbf-8af972d8c76c-0.avif\" alt=\"Output of the previous code cell\" />"
            ]
          },
          "execution_count": 4,
          "metadata": {},
          "output_type": "execute_result"
        }
      ],
      "source": [
        "from qiskit_mitigation.postselection.passes import (\n",
        "    AddSpectatorPostCircuitNonMarkovianErrorChecks,\n",
        "    AddSpectatorPreCircuitNonMarkovianErrorChecks,\n",
        ")\n",
        "\n",
        "qc_w_checks = PassManager(\n",
        "    [\n",
        "        AddSpectatorPreCircuitNonMarkovianErrorChecks(\n",
        "            coupling_map=backend.coupling_map\n",
        "        ),\n",
        "        AddSpectatorPostCircuitNonMarkovianErrorChecks(\n",
        "            coupling_map=backend.coupling_map\n",
        "        ),\n",
        "    ]\n",
        ").run(qc_w_checks)\n",
        "qc_w_checks.draw(\"mpl\", idle_wires=False, fold=-1)"
      ]
    },
    {
      "cell_type": "markdown",
      "id": "e7c604e0-acac-491b-ba25-03eebfff0ab9",
      "metadata": {},
      "source": [
        "<span id=\"part-2-demonstrate-non-markovian-error-checks-on-a-qpu-and-compare-postselection-strategies\" />\n",
        "\n",
        "### 第2部：QPUにおける非マルコフ型エラーチェックの実証と、ポストセレクション戦略の比較\n",
        "\n",
        "非マルコフ型エラーチェックとは何か、またそれがソフトウェアでどのように実装されているかについて理解できたところで、QPU上でより大規模な例を実行し、その動作を確認してみましょう。\n",
        "\n"
      ]
    },
    {
      "cell_type": "markdown",
      "id": "6e6d9173-b5e2-424c-9c94-595745a0679e",
      "metadata": {},
      "source": [
        "まず、回路を用意します。 回路の選択はそれほど重要ではありません。出力を簡単に確認できる回路であれば十分です。 ここでは、20キュービットの、 1D のミラーリングされたイジング回路を選択する。 この回路をハードウェア上で実行するのは容易ではなく、非マルコフ誤差の影響を受けるはずである。これにより、非マルコフ誤差チェックの有効性を実証する機会が得られる。 また、この対称的な構造により、ノイズの混入した期待値の計算結果を、厳密な値と簡単に比較することができます。\n",
        "\n",
        "この実験では、観測可能な期待値を算出しています。 観測量 $O$ の選択も重要ではないため、単に平均単一サイト磁化 $O=\\frac{1}{N}\\sum_i\\langle Z_i\\rangle$ を選択する。\n",
        "\n"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 5,
      "id": "04f01a93-874c-4b95-bfb9-8e29caa7eaff",
      "metadata": {},
      "outputs": [
        {
          "name": "stdout",
          "output_type": "stream",
          "text": [
            "two-qubit depth: 24\n",
            "total depth:     85\n"
          ]
        },
        {
          "data": {
            "text/plain": [
              "<Image src=\"/docs/images/addons/qiskit-mitigation/guides/postselection-with-non-markovian-error-checks/extracted-outputs/04f01a93-874c-4b95-bfb9-8e29caa7eaff-1.avif\" alt=\"Output of the previous code cell\" />"
            ]
          },
          "execution_count": 5,
          "metadata": {},
          "output_type": "execute_result"
        }
      ],
      "source": [
        "import numpy as np\n",
        "from qiskit.quantum_info import SparsePauliOp\n",
        "from qiskit.transpiler import CouplingMap\n",
        "\n",
        "num_qubits = 20\n",
        "initial_layout = initial_layout = [\n",
        "    124,\n",
        "    125,\n",
        "    126,\n",
        "    127,\n",
        "    128,\n",
        "    129,\n",
        "    130,\n",
        "    131,\n",
        "    132,\n",
        "    133,\n",
        "    134,\n",
        "    135,\n",
        "    139,\n",
        "    155,\n",
        "    154,\n",
        "    153,\n",
        "    152,\n",
        "    151,\n",
        "    150,\n",
        "    149,\n",
        "][:num_qubits]\n",
        "coupling_map = CouplingMap.from_line(num_qubits, bidirectional=False)\n",
        "\n",
        "# Num Trotter steps and `rx` angle\n",
        "num_steps = 6\n",
        "theta_rx = np.pi / 6\n",
        "\n",
        "# Average single-site magnetization\n",
        "observable = (\n",
        "    SparsePauliOp(\n",
        "        [\n",
        "            \"I\" * iq + \"Z\" + \"I\" * (num_qubits - iq - 1)\n",
        "            for iq in range(num_qubits)\n",
        "        ]\n",
        "    )\n",
        "    / num_qubits\n",
        ")\n",
        "\n",
        "# Create the Trotter circuit\n",
        "circuit = QuantumCircuit(num_qubits)\n",
        "edges = CouplingMap.from_line(num_qubits, bidirectional=False).get_edges()\n",
        "edges_a = [e for e in edges if e[0] % 2 == 0]\n",
        "edges_b = [e for e in edges if e[0] % 2 == 1]\n",
        "for _ in range(num_steps):\n",
        "    circuit.rx(theta_rx, [i for i in range(num_qubits)])\n",
        "    for edge in edges_a:\n",
        "        circuit.cz(edge[0], edge[1])\n",
        "    for edge in edges_b:\n",
        "        circuit.cz(edge[0], edge[1])\n",
        "circuit = circuit.compose(circuit.inverse())\n",
        "circuit.measure_all()\n",
        "\n",
        "# Transpile circuit to backend\n",
        "pm = generate_preset_pass_manager(\n",
        "    optimization_level=0, backend=backend, initial_layout=initial_layout\n",
        ")\n",
        "isa_circuit = pm.run(circuit)\n",
        "print(\n",
        "    f\"two-qubit depth: {isa_circuit.depth(lambda instr: instr.operation.num_qubits == 2)}\"\n",
        ")\n",
        "print(f\"total depth:     {isa_circuit.depth()}\")\n",
        "isa_circuit.draw(\"mpl\", fold=-1, idle_wires=False)"
      ]
    },
    {
      "cell_type": "markdown",
      "id": "bceae1e6-91d0-4b75-abc2-b5975154db48",
      "metadata": {},
      "source": [
        "ここで、7つの隣接するオブザーバー量子ビットを含め、すべての量子ビットに対して、回路実行前および実行後のチェックを追加します。 回路に考えられるすべてのチェックを追加しておけば、後処理の際にどのポストセレクション戦略を採用するかを決定することができます。\n",
        "\n"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 6,
      "id": "3f19dd87",
      "metadata": {},
      "outputs": [
        {
          "name": "stdout",
          "output_type": "stream",
          "text": [
            "cregs: [('meas', 20), ('meas_pre', 20), ('spec_pre', 7), ('meas_ps', 20), ('spec', 7), ('spec_ps', 7)]\n"
          ]
        }
      ],
      "source": [
        "# Use all non-Markovian error checks so we can compare all strategies in post-processing.\n",
        "combined = PassManager(\n",
        "    [\n",
        "        AddPreCircuitNonMarkovianErrorChecks(x_pulse_type=\"rx\"),\n",
        "        AddSpectatorPreCircuitNonMarkovianErrorChecks(\n",
        "            backend.coupling_map, x_pulse_type=\"rx\"\n",
        "        ),\n",
        "        AddPostCircuitNonMarkovianErrorChecks(x_pulse_type=\"rx\"),\n",
        "        AddSpectatorPostCircuitNonMarkovianErrorChecks(\n",
        "            backend.coupling_map, x_pulse_type=\"rx\"\n",
        "        ),\n",
        "    ]\n",
        ").run(isa_circuit)\n",
        "combined_isa = generate_preset_pass_manager(\n",
        "    optimization_level=0, backend=backend\n",
        ").run(combined)\n",
        "\n",
        "print(\"cregs:\", [(c.name, c.size) for c in combined_isa.cregs])"
      ]
    },
    {
      "cell_type": "markdown",
      "id": "13e29609-f543-47aa-9309-efc37016356b",
      "metadata": {},
      "source": [
        "このプリミティブ `Executor` を使用して、チェックを含む回路をサンプリングしてください。\n",
        "\n"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 7,
      "id": "433e7833",
      "metadata": {},
      "outputs": [
        {
          "name": "stdout",
          "output_type": "stream",
          "text": [
            "job id: d9badrm6hjac73ffe31g\n",
            "status: QUEUED\n"
          ]
        }
      ],
      "source": [
        "from qiskit_ibm_runtime import Executor\n",
        "from samplomatic.quantum_program import QuantumProgram\n",
        "\n",
        "# Raw and combined-checks circuits in a single 100k-shot program so they share calibration.\n",
        "program = QuantumProgram(shots=100_000)\n",
        "program.append_circuit_item(isa_circuit)  # raw circuit, no checks\n",
        "program.append_circuit_item(\n",
        "    combined_isa\n",
        ")  # combined pre + post non-Markovian error checks\n",
        "job = Executor(backend).run(program)\n",
        "print(f\"job id: {job.job_id()}\")\n",
        "print(f\"status: {job.status()}\")"
      ]
    },
    {
      "cell_type": "markdown",
      "id": "fb9b9c49-22c8-4187-905f-db8c50d5e63a",
      "metadata": {},
      "source": [
        "このクラ `PostSelector` スはチェックの構造 `creg` を把握しているため、サンプル内の各ビットを、それに対応する回路測定値に関連付けることができます。 この手法 `compute_mask` は、事後選択戦略（ノードまたはエッジ）と、検討対象とする非マルコフ誤差チェックの種類（回路前または回路後）が与えられれば、一連のサンプルに対してマスクを作成するために使用できる。 回路の前段戦略と後段戦略を単一のマスクに統合するには、それぞれのマスクを個別に計算し、その後、演算 `AND` を用いてそれらを組み合わせることができます。 ここでは、 を用いて、さまざまな戦略すべてについてマスクを作成 `PostSelector` し、利用可能な戦略のすべての組み合わせについて結果を確認します。\n",
        "\n",
        "以下の可視化図では、ポストセレクション戦略がポストセレクション率や回答の質にどのような影響を与えるかがわかります。 予想通り、ノードベースのポストセレクションを用いた戦略では、ポストセレクション率が最も低かった。 エッジベースの戦略はより寛容であり、選択後の生存率がより高くなる傾向があります。 ユーザーは、自身の問題に対してさまざまな戦略を試行し、ポストセレクション率とサンプリングされた分布の品質との適切なバランスを見極める必要があります。 この例では、回路の事前チェックおよび事後チェックを伴うノードベースのポストセレクションを用いた結果、ポストセレクション率 が の最良解が得られた `.028%`。 一部の戦略は、素の回路の性能を上回らなかった。 また、チェック項目を追加したものの、それらを一切使用しなかった場合（下のグラフの「選択なし」と表示された棒グラフ）が、最もパフォーマンスが劣ることがわかります。\n",
        "\n"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 8,
      "id": "43bc3829",
      "metadata": {},
      "outputs": [
        {
          "name": "stdout",
          "output_type": "stream",
          "text": [
            "raw circuit             : <m> = +0.5791   (kept 100000/100000)\n",
            "no selection            : <m> = +0.5211   (kept 100000/100000)\n",
            "pre-only (node)         : <m> = +0.6026   (kept   1357/100000)\n",
            "pre-only (edge)         : <m> = +0.5637   (kept  13029/100000)\n",
            "post-only (node)        : <m> = +0.5682   (kept   2254/100000)\n",
            "post-only (edge)        : <m> = +0.5265   (kept  27514/100000)\n",
            "pre=node + post=node    : <m> = +0.6393   (kept     28/100000)\n",
            "pre=node + post=edge    : <m> = +0.5799   (kept    442/100000)\n",
            "pre=edge + post=node    : <m> = +0.6006   (kept    317/100000)\n",
            "pre=edge + post=edge    : <m> = +0.5521   (kept   3836/100000)\n"
          ]
        },
        {
          "data": {
            "text/plain": [
              "<Image src=\"/docs/images/addons/qiskit-mitigation/guides/postselection-with-non-markovian-error-checks/extracted-outputs/43bc3829-1.avif\" alt=\"Output of the previous code cell\" />"
            ]
          },
          "metadata": {},
          "output_type": "display_data"
        }
      ],
      "source": [
        "import matplotlib.pyplot as plt\n",
        "from qiskit_mitigation.postselection import PostSelector\n",
        "\n",
        "selector = PostSelector.from_circuit(combined_isa, backend.coupling_map)\n",
        "\n",
        "exec_results = job.result()\n",
        "# The Executor returns numpy arrays keyed by classical-register name, so each result item is\n",
        "# already the {creg_name: ndarray} mapping that compute_mask expects.\n",
        "raw_bits = exec_results[0][\"meas\"]  # raw circuit, no checks\n",
        "bits = exec_results[1]  # combined pre+post circuit\n",
        "\n",
        "meas_bits = bits[\"meas\"]\n",
        "total = meas_bits[..., 0].size\n",
        "\n",
        "\n",
        "def magnetization(b, mask=None):\n",
        "    \"\"\"Average single-site magnetization over the (optionally masked) measurement bits.\"\"\"\n",
        "    kept = b if mask is None else b[mask]\n",
        "    return (1 - 2 * kept.astype(int)).mean()  # bit 0 -> +1, bit 1 -> -1\n",
        "\n",
        "\n",
        "rows = []\n",
        "\n",
        "\n",
        "def report(label, mask=None, b=meas_bits):\n",
        "    \"\"\"Record and print the magnetization and kept fraction for one selection strategy.\"\"\"\n",
        "    kept = b[..., 0].size if mask is None else int(mask.sum())\n",
        "    ev = magnetization(b, mask)\n",
        "    rows.append((label, ev, kept / total))\n",
        "    print(f\"{label:<24}: <m> = {ev:+.4f}   (kept {kept:>6}/{total})\")\n",
        "\n",
        "\n",
        "# Compare all possible postselection strategies\n",
        "report(\"raw circuit\", b=raw_bits)\n",
        "report(\"no selection\")\n",
        "for s in (\"node\", \"edge\"):\n",
        "    report(f\"pre-only ({s})\", selector.compute_mask(bits, s, mode=\"pre\"))\n",
        "for s in (\"node\", \"edge\"):\n",
        "    report(f\"post-only ({s})\", selector.compute_mask(bits, s, mode=\"post\"))\n",
        "for pre_s in (\"node\", \"edge\"):\n",
        "    for post_s in (\"node\", \"edge\"):\n",
        "        keep = selector.compute_mask(\n",
        "            bits, pre_s, mode=\"pre\"\n",
        "        ) & selector.compute_mask(bits, post_s, mode=\"post\")\n",
        "        report(f\"pre={pre_s} + post={post_s}\", keep)\n",
        "\n",
        "labels = [r[0] for r in rows]\n",
        "evs = [r[1] for r in rows]\n",
        "fracs = [r[2] for r in rows]\n",
        "x = range(len(labels))\n",
        "\n",
        "fig, ax1 = plt.subplots(figsize=(11, 5))\n",
        "\n",
        "# Bars: expectation value per selection strategy.\n",
        "bar_colors = [\"firebrick\"] + [\"steelblue\"] * (len(evs) - 1)\n",
        "ax1.bar(x, evs, color=bar_colors, alpha=0.85)\n",
        "ax1.axhline(1.0, color=\"green\", ls=\":\", lw=1.5, label=\"exact <m>\")\n",
        "ax1.axhline(rows[0][1], color=\"firebrick\", ls=\"--\", lw=1, label=\"raw <m>\")\n",
        "ax1.set_ylabel(\"<m>  (avg single-site magnetization)\", color=\"steelblue\")\n",
        "ax1.tick_params(axis=\"y\", labelcolor=\"steelblue\")\n",
        "ax1.set_xticks(list(x))\n",
        "ax1.set_xticklabels(labels, rotation=45, ha=\"right\")\n",
        "ax1.legend(loc=\"upper right\")\n",
        "\n",
        "# Line: fraction of shots retained (right axis).\n",
        "ax2 = ax1.twinx()\n",
        "ax2.plot(x, fracs, \"o-\", color=\"darkorange\")\n",
        "ax2.set_ylabel(\"kept fraction\", color=\"darkorange\")\n",
        "ax2.tick_params(axis=\"y\", labelcolor=\"darkorange\")\n",
        "ax2.set_ylim(0, 1.05)\n",
        "\n",
        "ax1.set_title(\"Selection strategy: expectation value vs shots retained\")\n",
        "fig.tight_layout()\n",
        "plt.show()"
      ]
    },
    {
      "cell_type": "markdown",
      "metadata": {},
      "id": "a1b8767d",
      "source": "© IBM Corp., 2017-2026"
    }
  ],
  "metadata": {
    "kernelspec": {
      "display_name": "Python 3",
      "language": "python",
      "name": "python3"
    },
    "language_info": {
      "codemirror_mode": {
        "name": "ipython",
        "version": 3
      },
      "file_extension": ".py",
      "mimetype": "text/x-python",
      "name": "python",
      "nbconvert_exporter": "python",
      "pygments_lexer": "ipython3",
      "version": "3"
    }
  },
  "nbformat": 4,
  "nbformat_minor": 5
}