Python Submodularの例

コード例 #1

ファイル: binarysm.py プロジェクト: pcreed/WPolyanna

def check_submodular(gamma):

    sm = Submodular()

    if sm.improves(gamma):
        print "%s is submodular." % str(gamma)
    else:
        print "%s is not submodular." % str(gamma)

コード例 #2

ファイル: test_submodular.py プロジェクト: pcreed/WPolyanna

 def setUp(self):
     self.sm = Submodular(2)
     self.proj3 = [MinMax(3,[[i]]) for i in [0,1,2]]
     self.f = MinMax(3,[[0,1],[0,2]])
     self.g = MinMax(3,[[0],[1],[2]])
     self.h = MinMax(3,[[0,1]])
     self.wop3 = WeightedOperation(3,2,self.proj3 + [self.f,self.g,self.h],
                                   [-1,-1,-1,1,1,1])

コード例 #3

ファイル: test_submodular.py プロジェクト: pcreed/WPolyanna

class TestSubmodular(unittest.TestCase):

    def setUp(self):
        self.sm = Submodular(2)
        self.proj3 = [MinMax(3,[[i]]) for i in [0,1,2]]
        self.f = MinMax(3,[[0,1],[0,2]])
        self.g = MinMax(3,[[0],[1],[2]])
        self.h = MinMax(3,[[0,1]])
        self.wop3 = WeightedOperation(3,2,self.proj3 + [self.f,self.g,self.h],
                                      [-1,-1,-1,1,1,1])

    def test_translations(self):
        self.assertEqual(self.sm.translations(2),[[-1,-1,1,1]])
                         
    def test_in_wclone(self):
        self.assertTrue(self.sm.in_wclone(self.wop3))
        self.assertTrue(self.wop3.in_wclone(self.sm,MinMax.clone(2)))