logistic-regression-copy.cc File Reference

#include "base/kaldi-common.h"
#include "util/common-utils.h"
#include "ivector/logistic-regression.h"

Include dependency graph for logistic-regression-copy.cc:

Go to the source code of this file.

Functions
int	main (int argc, char *argv[])

Function Documentation

main()

int main	(	int	argc,
		char *	argv[]
	)

Definition at line 26 of file logistic-regression-copy.cc.

References ParseOptions::GetArg(), KALDI_LOG, ParseOptions::NumArgs(), kaldi::PrintableWxfilename(), ParseOptions::PrintUsage(), ParseOptions::Read(), kaldi::ReadKaldiObject(), ParseOptions::Register(), LogisticRegression::ScalePriors(), and kaldi::WriteKaldiObject().

                                 {
  using namespace kaldi;
  typedef kaldi::int32 int32;
  try {
    const char *usage =
        "Copy a logistic-regression model, possibly changing the binary mode;\n"
        "also supports the --scale-priors option which can scale the prior probabilities\n"
        "the model assigns to different classes (e.g., you can remove the effect of\n"
        "unbalanced training data by scaling by the inverse of the class priors in the\n"
        "training data)\n"
        "Usage: logistic-regression-copy [options] <model-in> <model-out>\n"
        "e.g.: echo '[ 2.6 1.7 3.9 1.24 7.5 ]' | logistic-regression-copy --scale-priors=- \\\n"
        "  1.model scaled_priors.mdl\n";

    ParseOptions po(usage);

    bool binary = true;
    std::string scale_priors_rxfilename;

    po.Register("binary", &binary, "Write output in binary mode");
    po.Register("scale-priors", &scale_priors_rxfilename, "(extended) filename for file "
                "containing a vector of prior-scales (e.g. inverses of training priors)");

    po.Read(argc, argv);

    if (po.NumArgs() != 2) {
      po.PrintUsage();
      exit(1);
    }

    std::string model_rxfilename = po.GetArg(1),
        model_wxfilename = po.GetArg(2);


    LogisticRegression model;
    ReadKaldiObject(model_rxfilename, &model);

    if (scale_priors_rxfilename != "") {
      Vector<BaseFloat> prior_scales;
      ReadKaldiObject(scale_priors_rxfilename, &prior_scales);
      model.ScalePriors(prior_scales);
    }

    WriteKaldiObject(model, model_wxfilename, binary);

    KALDI_LOG << "Wrote model to " << PrintableWxfilename(model_wxfilename);
    return 0;
  } catch(const std::exception &e) {
    std::cerr << e.what();
    return -1;
  }
}