examples/Automata Learning.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "id": "d42f6057",
   "metadata": {},
   "source": [
    "Automata Learning\n",
    "================"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "0a42a66b",
   "metadata": {},
   "outputs": [],
   "source": [
    "from math import ceil\n",
    "from pyautospec import SpectralLearning"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "ddfc99d0",
   "metadata": {},
   "outputs": [],
   "source": [
    "alphabet = [\"a\", \"b\"]\n",
    "\n",
    "# instantiate a learner, here 3 is the length of the prefixes/suffixes used during learning\n",
    "learn = SpectralLearning(alphabet, 3)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "10596345",
   "metadata": {},
   "outputs": [],
   "source": [
    "# a function that counts 'b's\n",
    "def count_b(s):\n",
    "    return len([x for x in s if x == \"b\"])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "id": "9e9ed449",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "254e78013b2f445fab62be7de5e86d4e",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/15 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "No GPU/TPU found, falling back to CPU. (Set TF_CPP_MIN_LOG_LEVEL=0 and rerun for more info.)\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Singular values: [38.563854  4.563858]\n"
     ]
    },
    {
     "data": {
      "image/svg+xml": [
       "<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"no\"?>\n",
       "<!DOCTYPE svg PUBLIC \"-//W3C//DTD SVG 1.1//EN\"\n",
       " \"http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd\">\n",
       "<!-- Generated by graphviz version 8.0.3 (0)\n",
       " -->\n",
       "<!-- Pages: 1 -->\n",
       "<svg width=\"371pt\" height=\"202pt\"\n",
       " viewBox=\"0.00 0.00 371.00 202.00\" xmlns=\"http://www.w3.org/2000/svg\" xmlns:xlink=\"http://www.w3.org/1999/xlink\">\n",
       "<g id=\"graph0\" class=\"graph\" transform=\"scale(1 1) rotate(0) translate(4 198)\">\n",
       "<polygon fill=\"white\" stroke=\"none\" points=\"-4,4 -4,-198 367,-198 367,4 -4,4\"/>\n",
       "<!-- 0 -->\n",
       "<g id=\"node1\" class=\"node\">\n",
       "<title>0</title>\n",
       "<ellipse fill=\"none\" stroke=\"black\" cx=\"245\" cy=\"-22\" rx=\"18\" ry=\"18\"/>\n",
       "<ellipse fill=\"none\" stroke=\"black\" cx=\"245\" cy=\"-22\" rx=\"22\" ry=\"22\"/>\n",
       "<text text-anchor=\"middle\" x=\"245\" y=\"-18.3\" font-family=\"Times,serif\" font-size=\"14.00\">0</text>\n",
       "</g>\n",
       "<!-- 0&#45;&gt;0 -->\n",
       "<g id=\"edge5\" class=\"edge\">\n",
       "<title>0&#45;&gt;0</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M240.8,-43.81C240.4,-53.56 241.8,-62 245,-62 246.9,-62 248.16,-59.03 248.8,-54.57\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"252.32,-54.93 249.2,-44.81 245.33,-54.67 252.32,-54.93\"/>\n",
       "<text text-anchor=\"middle\" x=\"245\" y=\"-65.8\" font-family=\"Times,serif\" font-size=\"14.00\">a:1.0</text>\n",
       "</g>\n",
       "<!-- 0&#45;&gt;0 -->\n",
       "<g id=\"edge6\" class=\"edge\">\n",
       "<title>0&#45;&gt;0</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M237.82,-43.11C234.65,-61.1 237.05,-80 245,-80 251.34,-80 254.14,-68 253.42,-54.02\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"256.79,-53.65 252.18,-44.11 249.83,-54.45 256.79,-53.65\"/>\n",
       "<text text-anchor=\"middle\" x=\"245\" y=\"-83.8\" font-family=\"Times,serif\" font-size=\"14.00\">b:1.3</text>\n",
       "</g>\n",
       "<!-- 0a -->\n",
       "<g id=\"node5\" class=\"node\">\n",
       "<title>0a</title>\n",
       "<text text-anchor=\"middle\" x=\"336\" y=\"-18.3\" font-family=\"Times,serif\" font-size=\"14.00\">&#45;0.8</text>\n",
       "</g>\n",
       "<!-- 0&#45;&gt;0a -->\n",
       "<g id=\"edge3\" class=\"edge\">\n",
       "<title>0&#45;&gt;0a</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M267.09,-22C276.48,-22 287.8,-22 298.49,-22\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"298.3,-25.5 308.3,-22 298.3,-18.5 298.3,-25.5\"/>\n",
       "</g>\n",
       "<!-- 1 -->\n",
       "<g id=\"node2\" class=\"node\">\n",
       "<title>1</title>\n",
       "<ellipse fill=\"none\" stroke=\"black\" cx=\"118\" cy=\"-106\" rx=\"18\" ry=\"18\"/>\n",
       "<ellipse fill=\"none\" stroke=\"black\" cx=\"118\" cy=\"-106\" rx=\"22\" ry=\"22\"/>\n",
       "<text text-anchor=\"middle\" x=\"118\" y=\"-102.3\" font-family=\"Times,serif\" font-size=\"14.00\">1</text>\n",
       "</g>\n",
       "<!-- 1&#45;&gt;0 -->\n",
       "<g id=\"edge7\" class=\"edge\">\n",
       "<title>1&#45;&gt;0</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M136.91,-93.96C157.89,-79.87 192.85,-56.37 217.09,-40.08\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"218.78,-42.5 225.12,-34.01 214.87,-36.69 218.78,-42.5\"/>\n",
       "<text text-anchor=\"middle\" x=\"181.5\" y=\"-80.8\" font-family=\"Times,serif\" font-size=\"14.00\">b:0.3</text>\n",
       "</g>\n",
       "<!-- 1&#45;&gt;1 -->\n",
       "<g id=\"edge8\" class=\"edge\">\n",
       "<title>1&#45;&gt;1</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M113.8,-127.81C113.4,-137.56 114.8,-146 118,-146 119.9,-146 121.16,-143.03 121.8,-138.57\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"125.32,-138.93 122.2,-128.81 118.33,-138.67 125.32,-138.93\"/>\n",
       "<text text-anchor=\"middle\" x=\"118\" y=\"-149.8\" font-family=\"Times,serif\" font-size=\"14.00\">a:1.0</text>\n",
       "</g>\n",
       "<!-- 1&#45;&gt;1 -->\n",
       "<g id=\"edge9\" class=\"edge\">\n",
       "<title>1&#45;&gt;1</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M110.82,-127.11C107.65,-145.1 110.05,-164 118,-164 124.34,-164 127.14,-152 126.42,-138.02\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"129.79,-137.65 125.18,-128.11 122.83,-138.45 129.79,-137.65\"/>\n",
       "<text text-anchor=\"middle\" x=\"118\" y=\"-167.8\" font-family=\"Times,serif\" font-size=\"14.00\">b:0.7</text>\n",
       "</g>\n",
       "<!-- 1a -->\n",
       "<g id=\"node6\" class=\"node\">\n",
       "<title>1a</title>\n",
       "<text text-anchor=\"middle\" x=\"245\" y=\"-142.3\" font-family=\"Times,serif\" font-size=\"14.00\">0.8</text>\n",
       "</g>\n",
       "<!-- 1&#45;&gt;1a -->\n",
       "<g id=\"edge4\" class=\"edge\">\n",
       "<title>1&#45;&gt;1a</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M139.36,-112.52C157.79,-118.41 185.37,-127.24 207.59,-134.35\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"206.41,-137.97 217,-137.68 208.55,-131.3 206.41,-137.97\"/>\n",
       "</g>\n",
       "<!-- 0s -->\n",
       "<g id=\"node3\" class=\"node\">\n",
       "<title>0s</title>\n",
       "<text text-anchor=\"middle\" x=\"118\" y=\"-18.3\" font-family=\"Times,serif\" font-size=\"14.00\">&#45;0.8</text>\n",
       "</g>\n",
       "<!-- 0s&#45;&gt;0 -->\n",
       "<g id=\"edge1\" class=\"edge\">\n",
       "<title>0s&#45;&gt;0</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M144.9,-22C164.33,-22 191.04,-22 211.84,-22\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"211.7,-25.5 221.7,-22 211.7,-18.5 211.7,-25.5\"/>\n",
       "</g>\n",
       "<!-- 1s -->\n",
       "<g id=\"node4\" class=\"node\">\n",
       "<title>1s</title>\n",
       "<text text-anchor=\"middle\" x=\"27\" y=\"-102.3\" font-family=\"Times,serif\" font-size=\"14.00\">&#45;0.8</text>\n",
       "</g>\n",
       "<!-- 1s&#45;&gt;1 -->\n",
       "<g id=\"edge2\" class=\"edge\">\n",
       "<title>1s&#45;&gt;1</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M53.74,-106C63.37,-106 74.42,-106 84.57,-106\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"84.52,-109.5 94.52,-106 84.52,-102.5 84.52,-109.5\"/>\n",
       "</g>\n",
       "</g>\n",
       "</svg>\n"
      ],
      "text/plain": [
       "<graphviz.graphs.Digraph at 0x6a45604fa050>"
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# learn the count_a function\n",
    "counter = learn.learn(count_b)\n",
    "\n",
    "counter.diagram()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "id": "0ad7d705",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "0"
      ]
     },
     "execution_count": 5,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "ceil(counter(\"a\"))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "id": "e47c45c3",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "0"
      ]
     },
     "execution_count": 6,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "ceil(counter(\"aa\"))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "id": "046bc41d",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "1"
      ]
     },
     "execution_count": 7,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "ceil(counter(\"aabaaa\"))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "id": "cfd1c750",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "21"
      ]
     },
     "execution_count": 8,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "ceil(counter(\"aabaabbbbbbbbbbbbbbbbbbba\"))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "33092c00",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "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.10.10"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}