minor changes to tapex

kakugawa · Jan 24, 2023 · 571906b · 571906b
1 parent 0344a3c
commit 571906b
Showing 1 changed file with 0 additions and 31 deletions.
diff --git a/notebooks/tableqa/tapex_training.ipynb b/notebooks/tableqa/tapex_training.ipynb
@@ -22,37 +22,6 @@
     "\n",
     "tapex_reader.train()"
    ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# Testing the trained model\n",
-    "# Load the Table \n",
-    "data = {\"Actors\": [\"Brad Pitt\", \"Leonardo Di Caprio\",\n",
-    "                        \"George Clooney\"], \"Number of movies\": [\"87\", \"53\", \"69\"]}\n",
-    "print(pd.DataFrame.from_dict(data))"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "queries = [\"how many movies Brad Pitt acted in\", \"Name the actor who has been in 53 movies\"]\n",
-    "answers = tapex_reader.predict(data,queries)\n",
-    "print(\"answers\" , answers)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": []
   }
  ],
  "metadata": {