adds warning about needing to compute file with embeddings

2 years ago · 04f7e93c62
parent bd061ad2d4
commit 04f7e93c62
1 changed files with 1 additions and 1 deletions
--- a/examples/User_and_product_embeddings.ipynb
+++ b/examples/User_and_product_embeddings.ipynb
@ -39,7 +39,7 @@
    "import numpy as np\n",
    "from sklearn.model_selection import train_test_split\n",
    "\n",
-    "df = pd.read_csv('output/embedded_babbage_similarity_50k.csv', index_col=0)\n",
+    "df = pd.read_csv('output/embedded_babbage_similarity_50k.csv', index_col=0)  # note that you will need to generate this file to run the code below\n",
    "df['babbage_similarity'] = df.babbage_similarity.apply(eval).apply(np.array)\n",
    "X_train, X_test, y_train, y_test = train_test_split(df, df.Score, test_size = 0.2, random_state=42)\n",
    "\n",