def test_zero_known_item_scores(): train = get_random_coo_matrix().tocsr() predictions = np.random.random_sample(train.shape) r = BaseRecommender() safe = r._zero_known_item_scores(predictions, train) num_users, num_items = predictions.shape for u in xrange(num_users): for i in xrange(num_items): if i in train[u].indices: assert_less_equal(safe[u, i], 0) else: assert_equal(safe[u, i], predictions[u, i])
def test_zero_known_item_scores(): train = get_random_coo_matrix().tocsr() predictions = np.random.random_sample(train.shape) r = BaseRecommender() safe = r._zero_known_item_scores(predictions,train) num_users,num_items = predictions.shape for u in xrange(num_users): for i in xrange(num_items): if i in train[u].indices: assert_less_equal(safe[u,i],0) else: assert_equal(safe[u,i],predictions[u,i])