From 48f12a4052210a33e11154263a54aeea21abb954 Mon Sep 17 00:00:00 2001 From: david-cortes Date: Mon, 18 Jan 2021 19:58:34 +0200 Subject: [PATCH] correct nthreads passed to predict (#59) --- R/MatrixFactorizationRecommender.R | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/R/MatrixFactorizationRecommender.R b/R/MatrixFactorizationRecommender.R index 9314d1a..9a5d844 100644 --- a/R/MatrixFactorizationRecommender.R +++ b/R/MatrixFactorizationRecommender.R @@ -60,7 +60,7 @@ MatrixFactorizationRecommender = R6::R6Class( not_recommend = as(not_recommend, "RsparseMatrix") uids = rownames(user_embeddings) - indices = find_top_product(user_embeddings, item_embeddings, k, not_recommend, items_exclude, self$global_bias) + indices = find_top_product(user_embeddings, item_embeddings, k, not_recommend, items_exclude, glob_mean=self$global_bias) data.table::setattr(indices, "dimnames", list(uids, NULL)) data.table::setattr(indices, "ids", NULL)