From 90d3b2bd752aae78266bde0314e50be3e223e47d Mon Sep 17 00:00:00 2001 From: Evgeny Frolov Date: Sun, 3 Jun 2018 16:58:49 +0300 Subject: [PATCH] add note on possible caching --- polara/recommender/models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/polara/recommender/models.py b/polara/recommender/models.py index a4a8fde..c2e47ab 100644 --- a/polara/recommender/models.py +++ b/polara/recommender/models.py @@ -874,7 +874,7 @@ def slice_recommendations(self, test_data, shape, start, stop, test_users=None): scores = self.tensor_outer_at(1.0, v, w, slice_idx[1], slice_idx[2]) scores = np.add.reduceat(scores, np.r_[0, np.where(np.diff(slice_idx[0]))[0]+1]) - wt_flat = self.flatten_scores(w.T, self.flattener) + wt_flat = self.flatten_scores(w.T, self.flattener) # TODO cache result scores = np.tensordot(scores, wt_flat, axes=(2, 0)).dot(v.T) return scores, slice_idx