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",