From 1eb6619e7d78a8e4b9ac1750ef7ae6c61219af66 Mon Sep 17 00:00:00 2001 From: aaronpal Date: Tue, 27 Aug 2024 20:33:12 +0800 Subject: [PATCH] Correct variable used in pickle dump in `mind_utils.ipynb` Fixed an issue where the incorrect variable `word_dict` was being dumped to word_dict_all.pkl instead of `word_dict_all` in the `mind_utils.ipynb` notebook. Signed-off-by: aaron --- AUTHORS.md | 2 ++ examples/01_prepare_data/mind_utils.ipynb | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/AUTHORS.md b/AUTHORS.md index 1816f73e2..b70bfa644 100644 --- a/AUTHORS.md +++ b/AUTHORS.md @@ -52,6 +52,8 @@ To contributors: please add your name to the list when you submit a patch to the * **[Aaron He](https://github.com/AaronHeee)** * Reco utils of NCF * Deep dive notebook demonstrating the use of NCF +* **[Aaron Palpallatoc](https://github.com/ubergonmx)** + * Corrected variable in pickle dump in `mind_utils.ipynb` notebook * **[Abir Chakraborty](https://github.com/aeroabir)** * Self-Attentive Sequential Recommendation (SASRec) * Sequential Recommendation Via Personalized Transformer (SSEPT) diff --git a/examples/01_prepare_data/mind_utils.ipynb b/examples/01_prepare_data/mind_utils.ipynb index e03a3683d..7a2d81e6e 100644 --- a/examples/01_prepare_data/mind_utils.ipynb +++ b/examples/01_prepare_data/mind_utils.ipynb @@ -306,7 +306,7 @@ " pickle.dump(word_dict, f)\n", " \n", "with open(os.path.join(output_path, 'word_dict_all.pkl'), 'wb') as f:\n", - " pickle.dump(word_dict, f)" + " pickle.dump(word_dict_all, f)" ] }, {