Python topkの例

コード例 #1

ファイル: test_base.py プロジェクト: depengchen/mars

    def testTopk(self):
        raw = np.random.rand(20)
        a = tensor(raw, chunk_size=10)

        t = topk(a, 2)
        t = t.tiles()
        self.assertEqual(t.op.parallel_kind, 'tree')

コード例 #2

ファイル: test_base.py プロジェクト: qinxuye/mars

def test_topk():
    raw = np.random.rand(20)
    a = tensor(raw, chunk_size=10)

    t = topk(a, 2)
    t = tile(t)
    assert t.op.parallel_kind == 'tree'

    t = topk(a, 3)
    t = tile(t)
    assert t.op.parallel_kind == 'psrs'

    t = topk(sort(a), 3)
    t = tile(t)
    # k is less than 100
    assert t.op.parallel_kind == 'tree'

    with pytest.raises(ValueError):
        topk(a, 3, parallel_kind='unknown')

コード例 #3

ファイル: test_base.py プロジェクト: tomzhang/mars-1

    def testTopk(self):
        raw = np.random.rand(20)
        a = tensor(raw, chunk_size=10)

        t = topk(a, 2)
        t = t.tiles()
        self.assertEqual(t.op.parallel_kind, 'tree')

        t = topk(a, 3)
        t = t.tiles()
        self.assertEqual(t.op.parallel_kind, 'psrs')

        t = topk(sort(a), 3)
        t = t.tiles()
        # k is less than 100
        self.assertEqual(t.op.parallel_kind, 'tree')

        with self.assertRaises(ValueError):
            topk(a, 3, parallel_kind='unknown')