#include <opencv2/opencv.hpp>

#include <iostream>

using namespace cv;

using namespace std;

int main(int argc, char **argv)

{

    CvEM em; 

    Mat samples = Mat(3,2,CV_32F);

    samples.at<float>(0,0) = 1;

    samples.at<float>(1,0) = 2;

    samples.at<float>(3,0) = 3;

    CvEMParams params;

    params.nclusters = 2;

    Mat labels;

    em.train(samples, Mat(), params, &labels); 

    //< Write Out

    {

        FileStorage fs = FileStorage("./dump.em", FileStorage::WRITE);

        em.write(fs.fs, "EM");

    }

    //< Read In

    {

        FileStorage fs = FileStorage("./dump.em", FileStorage::READ);

        FileNode fileNode = fs["EM"];

        //< you need the const_cast to get this file compiled, this is not a trivial issue, 

        //< since all the read() functions require non-const pointer arguments

        em.read(const_cast<CvFileStorage*>(fileNode.fs), const_cast<CvFileNode*>(fileNode.node));

    }

    return 0;

}