Update fastText predict_proba output to conform to expected format

Author: gustavocidornelas

examples/text-classification/fasttext/fasttext.ipynb

@@ -625,7 +625,7 @@
     "\n",
     "import fasttext\n",
     "import pickle\n",
-    "\n",
+    "import numpy as np\n",
     "\n",
     "from pathlib import Path\n",
     "from typing import List\n",
@@ -646,7 +646,10 @@
     "    def predict_proba(self, input_data_df: pd.DataFrame):\n",
     "        \"\"\"Makes predictions with the model. Returns the class probabilities.\"\"\"\n",
     "        text_column = input_data_df.columns[0]\n",
-    "        return input_data_df[text_column].apply(self._predict_row)\n",
+    "        \n",
+    "        preds = input_data_df[text_column].apply(self._predict_row)\n",
+    "        \n",
+    "        return np.stack(preds.values)\n",
     "\n",
     "    def _predict_row(self, text: str) -> List[float]:\n",
     "        text = text.replace(\"\\n\",\" \")\n",