root/tests/arx_test.cpp @ 290

/*!
\file 
\brief Test of basic elements of the ARX class

See file \ref arx for mathematical background.

This class tests functions ARX::bayes (Bayes rule) ARX::structure_est and ARX::predictor_student

Untested functions: none.

 */

#include <estim/arx.h>
using namespace bdm;

int main() {
        // Setup model
        vec th ( "0.8 -0.3 0.4 0.01" );
        int ord=th.length(); //auxiliary variable
        double sqr=0.1;

        //Test constructor
        mat V0 = 0.00001*eye ( ord+1 ); V0 ( 0.0 ) = 1; //
        double nu0 = ord+5.0;

        ARX Ar;
        Ar.set_statistics(1, V0, nu0 );                 // Estimator
        const epdf& f_thr = Ar.posterior();          // refrence to posterior of the estimator

        //Test estimation
        int ndat = 100;                 // number of data records
        vec Yt ( ndat );                // Store generated data
        Yt.set_subvector ( 0,randn ( ord ) ); //initial values
        vec rgr ( ord );                // regressor
        vec Psi ( ord+1 );              // extended regressor

        //print moments of the prior distribution
        cout << "prior mean: " << f_thr.mean() <<endl;
        cout << "prior variance: " << f_thr.variance() <<endl;

        // cycle over time:
        for ( int t=ord; t<ndat; t++ ) {
                //Generate regressor
                for ( int j=0;j< ( ord );j++ ) {rgr ( j ) =Yt ( t-j-1 );}
                //model
                Yt ( t ) = th*rgr + sqr * NorRNG();

                Psi = concat ( Yt ( t ), rgr ); // Inefficient! Used for compatibility with Matlab!
                Ar.bayes ( Psi );             // Bayes rule

                // Build predictor
                mlstudent*      Pr = Ar.predictor_student ( );
                // Test similarity of likelihoods from the Bayes rule and the predictor
                cout << "BR log-lik: " << Ar._ll();
                cout <<" , predictor ll: " <<  Pr->evallogcond ( vec_1 ( Yt ( t ) ),rgr )  <<endl;
                delete Pr;
        }
        //print posterior moments
        cout << "posterior mean: " << f_thr.mean() <<endl;
        cout << "posterior variance: " << f_thr.variance() <<endl;

        // Test brute-froce structure estimation
        
        cout << "Structure estimation: " <<endl;
        cout <<Ar.structure_est ( egiw ( 1,V0,nu0 ) ) <<endl;
}