diff --git a/src/python/turicreate/test/test_recommender.py b/src/python/turicreate/test/test_recommender.py index 68ef000955..905b83c34e 100644 --- a/src/python/turicreate/test/test_recommender.py +++ b/src/python/turicreate/test/test_recommender.py @@ -1353,6 +1353,14 @@ def test_implicit(self): self._test_coreml_export(m4, ['135085','135038']) ''' + def test_recommend_from_interactions(self): + data=tc.SFrame({'userId':[1,1,1,2,2,2,3,3,3],'movieId':[10,11,12,10,13,14,10,11,14]}) + exclude_pairs=tc.SFrame({'movieId':[14]}) + recommendations=tc.SFrame({'movieId':[10]}) + model=tc.item_similarity_recommender.create(data,user_id='userId',item_id='movieId') + recommendations=model.recommend_from_interactions(observed_items=recommendations,exclude=exclude_pairs) + assert 14 not in recommendations['movieId'] + def test_compare_models(self): from turicreate.toolkits.recommender.util import compare_models diff --git a/src/python/turicreate/toolkits/recommender/util.py b/src/python/turicreate/toolkits/recommender/util.py index ed8469a258..6f4afa5d1f 100644 --- a/src/python/turicreate/toolkits/recommender/util.py +++ b/src/python/turicreate/toolkits/recommender/util.py @@ -1437,6 +1437,7 @@ def recommend_from_interactions( recommendations = self.recommend( users = users, new_observation_data = observed_items, + exclude = exclude, k = k, items = items, new_user_data = new_user_data,