test_roughly_optimized.py 文件源码

def test__compute_probabilities_loosely(self):
        b = il.RoughlyOptimized(self.lists, sample_num=3)
        is_success, p, minimum = b._compute_probabilities(
            self.lists,
            self.rankings,
        )
        assert is_success
        self.assert_almost_equal(p[0], 0.0)
        self.assert_almost_equal(p[1], 0.0)
        self.assert_almost_equal(p[2], 1.0)
        self.assert_almost_equal(b._lambdas[0], 0.5 - 1.0/3)
        self.assert_almost_equal(b._lambdas[1], 0.5 - 1.0/3 + 1.0/3 - 0.0)
        self.assert_almost_equal(
            minimum,
            np.sum(b._lambdas) + np.inner(p, b._sigmas),
        )
        _, _, minimum = b._compute_probabilities_loosely(
            self.lists,
            self.rankings,
            bias_weight=10.0,
        )
        self.assert_almost_equal(
            minimum,
            10.0 * np.sum(b._lambdas) + np.inner(p, b._sigmas),
        )