コード例 #1
0
ファイル: tests.py プロジェクト: florisvb/adskalman
    def test_learn_KPM(self):
        kpm=scipy.io.loadmat(pkg_resources.resource_filename(__name__,'kpm_learn_results'))

        y = kpm['y'].T # data vector is transposed from KPM
        F1 = kpm['F1']
        H1 = kpm['H1']
        Q1 = kpm['Q1']
        R1 = kpm['R1']
        initx1 = kpm['initx']
        initV1 = kpm['initV']
        max_iter = kpm['max_iter']
        F2, H2, Q2, R2, initx2, initV2, LL = adskalman.learn_kalman(y, F1, H1, Q1, R1, initx1, initV1, max_iter)
        assert numpy.allclose(F2,kpm['F2'])
        assert numpy.allclose(H2,kpm['H2'])
        assert numpy.allclose(Q2,kpm['Q2'])
        assert numpy.allclose(R2,kpm['R2'])
        assert numpy.allclose(initx2,kpm['initx2'])
        assert numpy.allclose(initV2,kpm['initV2'])
        #print numpy.ravel(LL),kpm['LL']
        assert numpy.allclose(LL,kpm['LL'])
コード例 #2
0
ファイル: tests.py プロジェクト: florisvb/adskalman
    def _test_kalman_murphy_EM(self): # temporarily disabled
        def returnC(orig):
            return self.C
        def returnQ(orig):
            return self.Q
        def returnR(orig):
            return self.R
        constr_fun_dict = {
            'C':returnC,
            #'Q':returnQ,
            #'R':returnR,
            }
        if 1:
            print '-'*80,'Murphy'
        Alearn, Clearn, Qlearn, Rlearn, initx2, initV2, LL = adskalman.learn_kalman(
            self.Y, self.Abad, self.C, self.Q, self.R, self.x0,
            self.P0, max_iter=500, verbose=2,
            constr_fun_dict=constr_fun_dict,
            thresh=1e-5,
            )

        xsmooth, Vsmooth, VVsmooth, loglik = adskalman.kalman_smoother(
            self.Y,self.A,self.C,self.Q,self.R,self.x0,self.P0,
            full_output=True)
        if 1:
            print 'best LL',loglik

            print
            print '*'*80,'Murphy'
            print 'A'
            print self.A
            print 'Alearn'
            print Alearn
            print 'Clearn'
            print Clearn
            print 'Qlearn'
            print Qlearn
            print 'Rlearn'
            print Rlearn
            print '*'*80,'Murphy'
コード例 #3
0
ファイル: tests.py プロジェクト: florisvb/adskalman
    def test_kalman__murphy1(self):
        def returnC(orig):
            return self.C
        def returnQ(orig):
            return self.Q
        def returnR(orig):
            return self.R
        constr_fun_dict = {
            'C':returnC,
            'Q':returnQ,
            'R':returnR,
            }
        print '-'*80,'Murphy'
        Alearn, Clearn, Qlearn, Rlearn, initx2, initV2, LL = adskalman.learn_kalman(
            self.Y, self.Abad, self.C, self.Q, self.R, self.x0,
            self.P0, max_iter=150, verbose=2,
            constr_fun_dict=constr_fun_dict,
            )

        xsmooth, Vsmooth, VVsmooth, loglik = adskalman.kalman_smoother(
            self.Y,self.A,self.C,self.Q,self.R,self.x0,self.P0,
            full_output=True)
        print 'best LL',loglik

        print
        print '*'*80,'Murphy'
        print 'A'
        print self.A
        print 'Alearn'
        print Alearn
        print 'Clearn'
        print Clearn
        print 'Qlearn'
        print Qlearn
        print 'Rlearn'
        print Rlearn
        print '*'*80,'Murphy'