[mlpack-svn] r16948 - mlpack/trunk/src/mlpack/methods/gmm
fastlab-svn at coffeetalk-1.cc.gatech.edu
fastlab-svn at coffeetalk-1.cc.gatech.edu
Tue Aug 5 08:57:21 EDT 2014
Author: michaelfox99
Date: Tue Aug 5 08:57:20 2014
New Revision: 16948
Log:
utility for converting legacy GMM XML files to new format
Added:
mlpack/trunk/src/mlpack/methods/gmm/gmm_convert_main.cpp
Added: mlpack/trunk/src/mlpack/methods/gmm/gmm_convert_main.cpp
==============================================================================
--- (empty file)
+++ mlpack/trunk/src/mlpack/methods/gmm/gmm_convert_main.cpp Tue Aug 5 08:57:20 2014
@@ -0,0 +1,63 @@
+/**
+ * @author Michael Fox
+ * @file gmm_convert_main.cpp
+ *
+ * This program converts an older GMM XML file to the new format.
+ */
+#include <mlpack/core.hpp>
+#include "gmm.hpp"
+
+
+using namespace mlpack;
+using namespace mlpack::gmm;
+using namespace mlpack::util;
+using namespace std;
+
+PROGRAM_INFO("Gaussian Mixture Model (GMM) file converter",
+ "This program takes a fitted GMM XML file from older MLPACK versions and "
+ "converts it to the current format.");
+
+PARAM_STRING_REQ("input_file", "File containing the fitted model ", "i");
+PARAM_STRING("output_file", "The file to write the model to (as XML).", "o",
+ "gmm.xml");
+
+int main(int argc, char* argv[])
+{
+ CLI::ParseCommandLine(argc, argv);
+ string inputFile = CLI::GetParam<string>("input_file");
+ //data::Load(CLI::GetParam<string>("input_file"), inputFile, true);
+ util::SaveRestoreUtility load;
+
+ if (!load.ReadFile(inputFile))
+ Log::Fatal << " Could not read file '" << inputFile << "'!\n";
+
+ size_t gaussians, dimensionality;
+ load.LoadParameter(gaussians, "gaussians");
+ load.LoadParameter(dimensionality, "dimensionality");
+ GMM<> gmm(gaussians, dimensionality);
+
+ load.LoadParameter(gmm.Weights(), "weights");
+
+ // We need to do a little error checking here.
+ if (gmm.Weights().n_elem != gmm.Gaussians())
+ {
+ Log::Fatal << "GMM::Load('" << inputFile << "'): file reports "
+ << gmm.Gaussians() << " gaussians but weights vector only contains "
+ << gmm.Weights().n_elem << " elements!" << std::endl;
+ }
+
+
+ for (size_t i = 0; i < gaussians; ++i)
+ {
+ std::stringstream o;
+ o << i;
+ std::string meanName = "mean" + o.str();
+ std::string covName = "covariance" + o.str();
+
+ load.LoadParameter(gmm.Component(i).Mean(), meanName);
+ load.LoadParameter(gmm.Component(i).Covariance(), covName);
+ }
+
+ gmm.Save(CLI::GetParam<string>("output_file"));
+
+}
More information about the mlpack-svn
mailing list