part1_example_qiskit.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "import qiskit\n",
    "from qiskit import quantum_info\n",
    "from qiskit.execute_function import execute\n",
    "from qiskit import BasicAer\n",
    "import numpy as np\n",
    "import pickle\n",
    "import json\n",
    "import os\n",
    "from collections import Counter\n",
    "from sklearn.metrics import mean_squared_error\n",
    "from typing import Dict, List\n",
    "import matplotlib.pyplot as plt\n",
    "\n",
    "#define utility functions\n",
    "\n",
    "def simulate(circuit: qiskit.QuantumCircuit) -> dict:\n",
    "    \"\"\"Simulate the circuit, give the state vector as the result.\"\"\"\n",
    "    backend = BasicAer.get_backend('statevector_simulator')\n",
    "    job = execute(circuit, backend)\n",
    "    result = job.result()\n",
    "    state_vector = result.get_statevector()\n",
    "    \n",
    "    histogram = dict()\n",
    "    for i in range(len(state_vector)):\n",
    "        population = abs(state_vector[i]) ** 2\n",
    "        if population > 1e-9:\n",
    "            histogram[i] = population\n",
    "    \n",
    "    return histogram\n",
    "\n",
    "\n",
    "def histogram_to_category(histogram):\n",
    "    \"\"\"This function take a histogram representations of circuit execution results, and process into labels as described in \n",
    "    the problem description.\"\"\"\n",
    "    assert abs(sum(histogram.values())-1)<1e-8\n",
    "    positive=0\n",
    "    for key in histogram.keys():\n",
    "        digits = bin(int(key))[2:].zfill(20)\n",
    "        if digits[-1]=='0':\n",
    "            positive+=histogram[key]\n",
    "        \n",
    "    return positive\n",
    "\n",
    "\n",
    "def count_gates(circuit: qiskit.QuantumCircuit) -> Dict[int, int]:\n",
    "    \"\"\"Returns the number of gate operations with each number of qubits.\"\"\"\n",
    "    counter = Counter([len(gate[1]) for gate in circuit.data])\n",
    "    #feel free to comment out the following two lines. But make sure you don't have k-qubit gates in your circuit\n",
    "    #for k>2\n",
    "    for i in range(2,20):\n",
    "        assert counter[i]==0\n",
    "        \n",
    "    return counter\n",
    "\n",
    "\n",
    "def image_mse(image1,image2):\n",
    "    # Using sklearns mean squared error:\n",
    "    # https://scikit-learn.org/stable/modules/generated/sklearn.metrics.mean_squared_error.html\n",
    "    return mean_squared_error(image1, image2)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "#load the mock data (for testing only)\n",
    "files=os.listdir(\"mock_data\")\n",
    "dataset=list()\n",
    "for file in files:\n",
    "    with open('mock_data/'+file, \"r\") as infile:\n",
    "        loaded = json.load(infile)\n",
    "        dataset.append(loaded)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "<matplotlib.image.AxesImage at 0x1318dcb80>"
      ]
     },
     "execution_count": 3,
     "metadata": {},
     "output_type": "execute_result"
    },
    {
     "data": {
      "image/png": "iVBORw0KGgoAAAANSUhEUgAAAPsAAAD4CAYAAAAq5pAIAAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjYuMCwgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy89olMNAAAACXBIWXMAAAsTAAALEwEAmpwYAAAQiElEQVR4nO3dXYyc9XXH8d/Z2Rd712vjxcY2xuUtkBZFwjRb2iqkJUJBhBuDFNFQKSIq7eYiSEHKRRGVGm4qoaokygWK5AQUp6KkqRKE06AG6iZCpK1jg1wwuAXXsbFdv2Js/MLuzu6eXuw4Wpt9zrOel52xz/cjrWb2OfPMHI/98zMz//k/f3N3Abj0dbW7AQDzg7ADSRB2IAnCDiRB2IEkuufzwXqtzxdoYD4f8qJgvb1hfWx5T1jv7q8W1qpj8b42EZZlU3F9qi8ezVk6cKaw9sH4gnDf7gMWP/jpD+N6QqM6rXEfm/WJayjsZnaXpG9Jqkj6rrs/Ht1+gQb0+3ZHIw95Sepe/Vth/Z2R1WF95ScPFtb27bwi3LfvSCWsd42HZZ25tvg/Gkm6b3hLYe3Fd3873PeKv4n/o9Kv3ojrCW32TYW1ul/Gm1lF0pOSPifpJkn3m9lN9d4fgNZq5D37rZJ2uvsudx+X9ANJ65rTFoBmayTsqyXtnfH7vtq2c5jZiJltNbOtVY018HAAGtHyT+Pdfb27D7v7cI/6Wv1wAAo0Evb9ktbM+P2q2jYAHaiRsG+RdIOZXWtmvZK+IGljc9oC0Gx1D725+4SZPSTpZ5oeenva3d9sWmeJ7PzzeGjt+T99Iqz/+4fXFdbWXP9euO+dwRj9XGw83R/Wd46tLKz95Jbvhvt+euThsH7jr8IyztPQOLu7vyDphSb1AqCF+LoskARhB5Ig7EAShB1IgrADSRB2IIl5nc+O2a38z8mwvuXz8RTYd8cvL6wdqi4J9335VDyXvlIyob2vZEJ81Yun0G4evTLct39X3BsuDEd2IAnCDiRB2IEkCDuQBGEHkiDsQBIMvXWAhYfiUyJXPf5rioa/Biuj4b79ldaeKuzERPEU2GhYTpK6Tze7m9w4sgNJEHYgCcIOJEHYgSQIO5AEYQeSIOxAEoyzd4CuE8XLGkvS3QM7w/qTx/6wsFY2zl421t1j8fTbsvqZqeJpqqen4hWCBg6WrBeNC8KRHUiCsANJEHYgCcIOJEHYgSQIO5AEYQeSYJy9A5y4eVlY3zJ2RVg/NVk8Xj02NVRXT2dNuTW0/7HqQGHtqt73w30PfCYeZx/8x7paSquhsJvZbkknJU1KmnD34WY0BaD5mnFk/4y7H23C/QBoId6zA0k0GnaX9KKZvWpmI7PdwMxGzGyrmW2tqrXnOwNQrNGX8be5+34zu0LSS2b23+7+8swbuPt6SeslabENeYOPB6BODR3Z3X1/7fKwpOck3dqMpgA0X91hN7MBMxs8e13SnZK2N6sxAM3VyMv4FZKeM7Oz9/MP7v4vTekqmwePhOXV3cfDel9X8Xnjy+abl42jL+s9FdbfHYvH8Qe7i+fTj5acD//jN+4P67wnvDB1h93dd0m6uYm9AGghht6AJAg7kARhB5Ig7EAShB1IgimuHeCyL8VrEx//5cKwvqCrWlg7NRGfrvm6hfGw347TV4b17q54aK+RYcG3d1wV1m9QPDSHc3FkB5Ig7EAShB1IgrADSRB2IAnCDiRB2IEkGGfvABMHD4X1F47Hkws/vfjtwtpPj8X7Hq0OhvWrF8bnEj04tiSsR1Nopzw+1qz6RVjGBeLIDiRB2IEkCDuQBGEHkiDsQBKEHUiCsANJMM5+EThe7Q/ra7qPFdZ+b/Gvw33fOhPPVy+zqDte0qs6VSmsjU71hPsO7opPY82ppC8MR3YgCcIOJEHYgSQIO5AEYQeSIOxAEoQdSIJx9ovAK3uuC+t/smxzYe39iYFw37UD74b1srHwMtVgWeajE4vCfbs+LD4fviTFZ53H+UqP7Gb2tJkdNrPtM7YNmdlLZvZO7XJpa9sE0Ki5vIz/nqS7ztv2iKRN7n6DpE213wF0sNKwu/vLks7/PuY6SRtq1zdIuqe5bQFotnrfs69w9wO16wclrSi6oZmNSBqRpAWKv+MNoHUa/jTe3V3BnAR3X+/uw+4+3KN4kUEArVNv2A+Z2SpJql0ebl5LAFqh3rBvlPRA7foDkp5vTjsAWqX0PbuZPSvpdknLzGyfpK9LelzSD83sQUl7JN3Xyiazq47Gf02jXjwWXvXi+eSSNFly7vYyZWusR2vHX9nzfrjv5oVr62kJBUrD7u73F5TuaHIvAFqIr8sCSRB2IAnCDiRB2IEkCDuQBFNcLwKf/Z0dYf3IxOLCWtnQ26FqvOTylb3x8NgCi6eh7q0OFdaG+uJTRR/663hY74p1YRnn4cgOJEHYgSQIO5AEYQeSIOxAEoQdSIKwA0kwzn4RuG7h0bB+YrL4dF9Lu0+H+/Z3jYf1Lk2F9WgcXZIqwcLKe6uXh/v+2cf+I6z/tHt5WPeJibCeDUd2IAnCDiRB2IEkCDuQBGEHkiDsQBKEHUiCcfYOMPXHt4T1/q5/C+snJxcU1nq64rHmd8fise6KxePsSyofhvWh7uI560cmBsN9y05TffLeT4b1Rf9UvJR1RhzZgSQIO5AEYQeSIOxAEoQdSIKwA0kQdiAJxtk7wJGbF4b1vmDZY0k6MVm8/5nJvnDfsnH0svnw1/TGc+2PB3Pty3orG2c//rH4nPiLwmo+pUd2M3vazA6b2fYZ2x4zs/1mtq32c3dr2wTQqLm8jP+epLtm2f5Nd19b+3mhuW0BaLbSsLv7y5KOzUMvAFqokQ/oHjKz12sv85cW3cjMRsxsq5ltrWqsgYcD0Ih6w/5tSddLWivpgKQnim7o7uvdfdjdh3sUfyADoHXqCru7H3L3SXefkvQdSbc2ty0AzVZX2M1s1Yxf75W0vei2ADpD6Ti7mT0r6XZJy8xsn6SvS7rdzNZKckm7JX25dS1e+s6sLD63uiQNdMWfdYxO9RTWBiujde8rSZdVzoT1w8Ha8JI05VZcU3FNkj6xcG9YnxiInzecqzTs7n7/LJufakEvAFqIr8sCSRB2IAnCDiRB2IEkCDuQBFNcO0B1KJ7KubZvX1jfM7assLaoZOjtqt542kM0RVWSTkzE9SXdxUN31al4iuqxyXiS6vjKeOovzsWRHUiCsANJEHYgCcIOJEHYgSQIO5AEYQeSYJy9A3QvHg/rvzhzY933vabnvbC+Z3x5WB/1+J9INI4uSVUvHkvvr8R/7n3jQ2G9f2m8XDTOxZEdSIKwA0kQdiAJwg4kQdiBJAg7kARhB5JgnL0DLB6Mx6pX97wf1pd3f1BY2zm2Mtx3suR0zksq8Vj2qckFYT3SpfhU0KdKlnRe3B/P1ce5OLIDSRB2IAnCDiRB2IEkCDuQBGEHkiDsQBKMs3eASlc83ry8UjyOLkmvnP543Y+9pGRJ5rGSJZ2j+eqS1GPF58Sv2FS8b1d8Pv1FvfF8eJyr9MhuZmvM7Odm9paZvWlmX61tHzKzl8zsndrl0ta3C6Bec3kZPyHpa+5+k6Q/kPQVM7tJ0iOSNrn7DZI21X4H0KFKw+7uB9z9tdr1k5J2SFotaZ2kDbWbbZB0T4t6BNAEF/Se3cyukXSLpM2SVrj7gVrpoKQVBfuMSBqRpAWK1wUD0Dpz/jTezBZJ+pGkh939nE+M3N2l2Wc1uPt6dx929+EexRMbALTOnMJuZj2aDvoz7v7j2uZDZraqVl8l6XBrWgTQDKUv483MJD0laYe7f2NGaaOkByQ9Xrt8viUdJvDheDy8tcAmwvqkF/+f3dMV79tl8bDfmcnesB4NrU3ff/Hw2mTJsF6ZJb3x9NuTDd37pWcu79k/JemLkt4ws221bY9qOuQ/NLMHJe2RdF9LOgTQFKVhd/dXpMIzHNzR3HYAtApflwWSIOxAEoQdSIKwA0kQdiAJprh2gPHx+K/hsq76p3KWjdGXTWHtL3nssiWd+4Jx+LLpsaMlvfWWTIHFuTiyA0kQdiAJwg4kQdiBJAg7kARhB5Ig7EASjLN3gPEz8Xjy7oklYX2wUrx0cdmSzI0qG4eP5ruXjbNPeWt7z4YjO5AEYQeSIOxAEoQdSIKwA0kQdiAJwg4kwTh7B6gcjc/NPthVPI4uST3BnPW+kvPClykbC68oXnb5xOTCwtqCrmpdPZ010B2P8b/X0L1fejiyA0kQdiAJwg4kQdiBJAg7kARhB5Ig7EASc1mffY2k70taIcklrXf3b5nZY5L+QtKR2k0fdfcXWtXopaw7Xma8VDU4d3uP4vPGj3rZ2vDxWPjy7ngV9NFq8f2fmOoP9x2biv95Rmu/46Pm8qWaCUlfc/fXzGxQ0qtm9lKt9k13/7vWtQegWeayPvsBSQdq10+a2Q5Jq1vdGIDmuqD37GZ2jaRbJG2ubXrIzF43s6fNbGnBPiNmttXMtlY11li3AOo257Cb2SJJP5L0sLt/IOnbkq6XtFbTR/4nZtvP3de7+7C7D/eor/GOAdRlTmE3sx5NB/0Zd/+xJLn7IXefdPcpSd+RdGvr2gTQqNKwm5lJekrSDnf/xoztq2bc7F5J25vfHoBmmcun8Z+S9EVJb5jZttq2RyXdb2ZrNT0ct1vSl1vQXwqLd8X13pJppFf3HimsLS6ZHnv7wvi+902cCusrKsVTWCXp/akDhbXdE/HU3r3Vy8P6Px+7OazjXHP5NP4VadaTjzOmDlxE+AYdkARhB5Ig7EAShB1IgrADSRB2IAlOJd0Blv/s12F92yNrwvqTO28vrHU/MxTuO/TKvrDuC0u+4lyJjxf2wel4/wZM7P+/lt33pYgjO5AEYQeSIOxAEoQdSIKwA0kQdiAJwg4kYe6NLel7QQ9mdkTSnhmblkk6Om8NXJhO7a1T+5LorV7N7O1qd18+W2Few/6RBzfb6u7DbWsg0Km9dWpfEr3Va75642U8kARhB5Jod9jXt/nxI53aW6f2JdFbvealt7a+Zwcwf9p9ZAcwTwg7kERbwm5md5nZ/5jZTjN7pB09FDGz3Wb2hpltM7Otbe7laTM7bGbbZ2wbMrOXzOyd2uWsa+y1qbfHzGx/7bnbZmZ3t6m3NWb2czN7y8zeNLOv1ra39bkL+pqX523e37ObWUXS25I+K2mfpC2S7nf3t+a1kQJmtlvSsLu3/QsYZvZHkk5J+r67f6K27W8lHXP3x2v/US5197/skN4ek3Sq3ct411YrWjVzmXFJ90j6ktr43AV93ad5eN7acWS/VdJOd9/l7uOSfiBpXRv66Hju/rKkY+dtXidpQ+36Bk3/Y5l3Bb11BHc/4O6v1a6flHR2mfG2PndBX/OiHWFfLWnvjN/3qbPWe3dJL5rZq2Y20u5mZrHC3c+uqXRQ0op2NjOL0mW859N5y4x3zHNXz/LnjeIDuo+6zd1/V9LnJH2l9nK1I/n0e7BOGjud0zLe82WWZcZ/o53PXb3LnzeqHWHfL2nmGRSvqm3rCO6+v3Z5WNJz6rylqA+dXUG3dnm4zf38Rict4z3bMuPqgOeuncuftyPsWyTdYGbXmlmvpC9I2tiGPj7CzAZqH5zIzAYk3anOW4p6o6QHatcfkPR8G3s5R6cs4120zLja/Ny1fflzd5/3H0l3a/oT+f+V9Fft6KGgr+sk/Vft58129ybpWU2/rKtq+rONByVdLmmTpHck/aukoQ7q7e8lvSHpdU0Ha1WbertN0y/RX5e0rfZzd7ufu6CveXne+LoskAQf0AFJEHYgCcIOJEHYgSQIO5AEYQeSIOxAEv8PE4DsMh8Hn6sAAAAASUVORK5CYII=",
      "text/plain": [
       "<Figure size 432x288 with 1 Axes>"
      ]
     },
     "metadata": {
      "needs_background": "light"
     },
     "output_type": "display_data"
    }
   ],
   "source": [
    "#load the actual hackthon data (fashion-mnist)\n",
    "images=np.load('data/images.npy')\n",
    "labels=np.load('data/labels.npy')\n",
    "#you can visualize it\n",
    "plt.imshow(images[1100])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [],
   "source": [
    "#submission to part 1, you should make this into a .py file\n",
    "\n",
    "n=len(dataset)\n",
    "mse=0\n",
    "gatecount=0\n",
    "\n",
    "# Functions 'encode' and 'decode' are dummy.\n",
    "def encode(image):\n",
    "    q = qiskit.QuantumRegister(3)\n",
    "    circuit = qiskit.QuantumCircuit(q)\n",
    "    if image[0][0]==0:\n",
    "        circuit.rx(np.pi,0)\n",
    "    return circuit\n",
    "\n",
    "def decode(histogram):\n",
    "    if 1 in histogram.keys():\n",
    "        image=[[0,0],[0,0]]\n",
    "    else:\n",
    "        image=[[1,1],[1,1]]\n",
    "    return image\n",
    "\n",
    "def run_part1(image):\n",
    "    #encode image into a circuit\n",
    "    circuit=encode(image)\n",
    "\n",
    "    #simulate circuit\n",
    "    histogram=simulate(circuit)\n",
    "\n",
    "    #reconstruct the image\n",
    "    image_re=decode(histogram)\n",
    "\n",
    "    return circuit,image_re"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "1.0\n"
     ]
    }
   ],
   "source": [
    "#how we grade your submission\n",
    "\n",
    "n=len(dataset)\n",
    "mse=0\n",
    "gatecount=0\n",
    "\n",
    "for data in dataset:\n",
    "    #encode image into circuit\n",
    "    circuit,image_re=run_part1(data['image'])\n",
    "    \n",
    "    #count the number of 2qubit gates used\n",
    "    gatecount+=count_gates(circuit)[2]\n",
    "    \n",
    "    #calculate mse\n",
    "    mse+=image_mse(data['image'],image_re)\n",
    "    \n",
    "#fidelity of reconstruction\n",
    "f=1-mse\n",
    "gatecount=gatecount/n\n",
    "\n",
    "#score for part1 \n",
    "print(f*(0.999**gatecount))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "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.9.15 (main, Oct 11 2022, 22:27:25) \n[Clang 14.0.0 (clang-1400.0.29.102)]"
  },
  "orig_nbformat": 4,
  "vscode": {
   "interpreter": {
    "hash": "397704579725e15f5c7cb49fe5f0341eb7531c82d19f2c29d197e8b64ab5776b"
   }
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}