| 1 | #include "constructionModelesLassoMLE.h" |
| 2 | #include "ioutils.h" |
| 3 | |
| 4 | int main(int argc, char** argv) |
| 5 | { |
| 6 | // read dimensions |
| 7 | const Int nbDims = 5; |
| 8 | Int* dimensions = readArray_int("dimensions",&nbDims,1); |
| 9 | mwSize n = dimensions[0]; |
| 10 | mwSize p = dimensions[1]; |
| 11 | mwSize m = dimensions[2]; |
| 12 | mwSize k = dimensions[3]; |
| 13 | mwSize L = dimensions[4]; |
| 14 | free(dimensions); |
| 15 | mwSize lengthOne = 1; |
| 16 | |
| 17 | //////////// |
| 18 | // INPUTS // |
| 19 | //////////// |
| 20 | |
| 21 | // phiInit |
| 22 | const mwSize dimPhiInit[] = {p, m, k}; |
| 23 | Real* phiInit = readArray_real("phiInit",dimPhiInit,3); |
| 24 | |
| 25 | // rhoInit |
| 26 | const mwSize dimRhoInit[] = {m, m, k}; |
| 27 | Real* rhoInit = readArray_real("rhoInit",dimRhoInit,3); |
| 28 | |
| 29 | // piInit |
| 30 | Real* piInit = readArray_real("piInit",&k,1); |
| 31 | |
| 32 | // gamInit |
| 33 | const mwSize dimGamInit[] = {n, k}; |
| 34 | Real* gamInit = readArray_real("gamInit",dimGamInit,2); |
| 35 | |
| 36 | // min number of iterations |
| 37 | Int* pmini = readArray_int("mini",&lengthOne,1); |
| 38 | Int mini = *pmini; |
| 39 | free(pmini); |
| 40 | |
| 41 | // max number of iterations |
| 42 | Int* pmaxi = readArray_int("maxi",&lengthOne,1); |
| 43 | Int maxi = *pmaxi; |
| 44 | free(pmaxi); |
| 45 | |
| 46 | // gamma |
| 47 | Real* pgamma = readArray_real("gamma",&lengthOne,1); |
| 48 | Real gamma = *pgamma; |
| 49 | free(pgamma); |
| 50 | |
| 51 | // lambda |
| 52 | Real* glambda = readArray_real("glambda",&L,1); |
| 53 | |
| 54 | // X |
| 55 | const mwSize dimX[] = {n, p}; |
| 56 | Real* X = readArray_real("X",dimX,2); |
| 57 | |
| 58 | // Y |
| 59 | const mwSize dimY[] = {n, m}; |
| 60 | Real* Y = readArray_real("Y",dimY,2); |
| 61 | |
| 62 | // seuil |
| 63 | Real* pseuil = readArray_real("seuil",&lengthOne,1); |
| 64 | Real seuil = *pseuil; |
| 65 | free(pseuil); |
| 66 | |
| 67 | // tau |
| 68 | Real* ptau = readArray_real("tau",&lengthOne,1); |
| 69 | Real tau = *ptau; |
| 70 | free(ptau); |
| 71 | |
| 72 | // A1 |
| 73 | const mwSize dimA[] = {p, m+1, L}; |
| 74 | Int* A1 = readArray_int("A1",dimA,3); |
| 75 | |
| 76 | // A2 |
| 77 | Int* A2 = readArray_int("A2",dimA,3); |
| 78 | |
| 79 | ///////////// |
| 80 | // OUTPUTS // |
| 81 | ///////////// |
| 82 | |
| 83 | // phi |
| 84 | const mwSize dimPhi[] = {dimPhiInit[0], dimPhiInit[1], dimPhiInit[2], L}; |
| 85 | Real* phi = (Real*)malloc(dimPhi[0]*dimPhi[1]*dimPhi[2]*dimPhi[3]*sizeof(Real)); |
| 86 | |
| 87 | // rho |
| 88 | const mwSize dimRho[] = {dimRhoInit[0], dimRhoInit[1], dimRhoInit[2], L}; |
| 89 | Real* rho = (Real*)malloc(dimRho[0]*dimRho[1]*dimRho[2]*dimRho[3]*sizeof(Real)); |
| 90 | |
| 91 | // pi |
| 92 | const mwSize dimPi[] = {k, L}; |
| 93 | Real* pi = (Real*)malloc(dimPi[0]*dimPi[1]*sizeof(Real)); |
| 94 | |
| 95 | // lvraisemblance |
| 96 | const mwSize dimLvraisemblance[] = {L, 2}; |
| 97 | Real* lvraisemblance = (Real*)malloc(dimLvraisemblance[0]*dimLvraisemblance[1]*sizeof(Real)); |
| 98 | |
| 99 | ///////////////////////////////////////// |
| 100 | // Call to constructionModelesLassoMLE // |
| 101 | ///////////////////////////////////////// |
| 102 | |
| 103 | constructionModelesLassoMLE( |
| 104 | phiInit,rhoInit,piInit,gamInit,mini,maxi,gamma,glambda,X,Y,seuil,tau,A1,A2, |
| 105 | phi,rho,pi,lvraisemblance, |
| 106 | n,p,m,k,L); |
| 107 | |
| 108 | free(phiInit); |
| 109 | free(rhoInit); |
| 110 | free(piInit); |
| 111 | free(gamInit); |
| 112 | free(X); |
| 113 | free(Y); |
| 114 | free(A1); |
| 115 | free(A2); |
| 116 | free(glambda); |
| 117 | |
| 118 | // Compare to reference outputs |
| 119 | Real* ref_phi = readArray_real("phi",dimPhi,4); |
| 120 | compareArray_real("phi", phi, ref_phi, dimPhi[0]*dimPhi[1]*dimPhi[2]*dimPhi[3]); |
| 121 | free(phi); |
| 122 | free(ref_phi); |
| 123 | |
| 124 | // rho |
| 125 | Real* ref_rho = readArray_real("rho",dimRho,4); |
| 126 | compareArray_real("rho", rho, ref_rho, dimRho[0]*dimRho[1]*dimRho[2]*dimRho[3]); |
| 127 | free(rho); |
| 128 | free(ref_rho); |
| 129 | |
| 130 | // pi |
| 131 | Real* ref_pi = readArray_real("pi",dimPi,2); |
| 132 | compareArray_real("pi", pi, ref_pi, dimPi[0]*dimPi[1]); |
| 133 | free(pi); |
| 134 | free(ref_pi); |
| 135 | |
| 136 | // lvraisemblance |
| 137 | Real* ref_lvraisemblance = readArray_real("lvraisemblance",dimLvraisemblance,2); |
| 138 | compareArray_real("lvraisemblance", lvraisemblance, ref_lvraisemblance, dimLvraisemblance[0]*dimLvraisemblance[1]); |
| 139 | free(lvraisemblance); |
| 140 | free(ref_lvraisemblance); |
| 141 | |
| 142 | return 0; |
| 143 | } |