int main(int argc, char **argv) { std::vector<Demonstration> demos; Demonstration demo = Demonstration(nbVar,nbData); std::vector<std::string> vars; vars.push_back("t"); vars.push_back("x"); vars.push_back("y"); vars.push_back("z"); demo.getDatapoints().loadFromFile("../../data/data_txyz.txt"); demo.getDatapoints().setVarNames(vars); cout << endl << "Data is: " << demo.getDatapoints().getData() << endl; cout << endl << "nbVar is: " << demo.getDatapoints().getNumVARS() << endl; cout << endl << "nbDatapoints is: " << demo.getDatapoints().getNumPOINTS() << endl; cout << "\nPress enter to continue..." << endl; getchar(); demos.push_back(demo); GMM_Model *gmm; gmm = new GMM_Model(demos,nbStates); gmm->setVARSNames(vars); cout << "\n Number of EM iterations: " << gmm->EM_learn(); for (int i=0;i<nbStates;i++){ cout << "\n Mu_" << i << " = \n" << gmm->getMU(i); cout << "\n Sigma_" << i << " = \n" << gmm->getSIGMA(i); } //Save GMM in files GMM_priors.txt, GMM_mu.txt, GMM_sigma.txt and GMM_vars.txt gmm->saveInFiles(); cout << "\nPress enter to continue..." << endl; getchar(); //Loading the GMM model from files... GMM_Model *gmm2 = new GMM_Model("GMM_priors.txt", "GMM_mu.txt", "GMM_sigma.txt", "GMM_vars.txt"); for (int i=0;i<nbStates;i++) gmm2->setMU(i, gmm2->getMU(i)*10.0); for (int i=0;i<nbStates;i++) cout << "\n Mu_" << i << " = \n" << gmm2->getMU(i); return 0; }