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])
コード例 #2
0
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])