diff --git a/examples/User_and_product_embeddings.ipynb b/examples/User_and_product_embeddings.ipynb index 9dacfb82..565ccd28 100644 --- 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",