lr_classify.cpp

/********************************************************************
* Logit Model (Softmax Regression) V0.20
* Implemented by Rui Xia(rxiacn@gmail.com)
* Last updated on 2013-05-04. 
*********************************************************************/
#include <cstdlib>
#include <iostream>
#include <cstring>
#include "LR.h"

using namespace std;


void print_help()
{
	cout << "\nOpenPR-LR classification module, " << VERSION << ", " << VERSION_DATE << "\n\n"
		<< "usage: lr_classify [options] testing_file model_file output_file\n\n"
		<< "options: -h        -> help\n"
		<< "         -f [0..2] -> 0: only output class label (default)\n"
		<< "                   -> 1: output class label with log-likelihood (weighted sum)\n"
		<< "                   -> 2: output class label with soft probability\n"
		<< endl;
}

void read_parameters(int argc, char *argv[], char *testing_file, char *model_file, 
						char *output_file, int *output_format) 
{
	// set default options
	*output_format = 0;
	int i;
	for (i = 1; (i<argc) && (argv[i])[0]=='-'; i++)
	{
		switch ((argv[i])[1]) {
			case 'h':
				print_help();
				exit(0);
			case 'f':
				*output_format = atoi(argv[++i]);
				break;
			default:
				cout << "Unrecognized option: " << argv[i] << "!" << endl;
				print_help();
				exit(0);
		}
	}
	
	if ((i+2)>=argc)
	{
		cout << "Not enough parameters!" << endl;
		print_help();
		exit(0);
	}
	strcpy(testing_file, argv[i]);
	strcpy(model_file, argv[i+1]);
	strcpy(output_file, argv[i+2]);
}

int lr_classify(int argc, char *argv[])
{
	char testing_file[200];
	char model_file[200];
	char output_file[200];
	int output_format;
	read_parameters(argc, argv, testing_file, model_file, output_file, &output_format);
    LR Logit;
	Logit.load_model(model_file);
	float acc = Logit.classify_testing_file(testing_file, output_file, output_format);
	cout << "Accuracy: " << acc << endl;
	return 0;
}

int main(int argc, char *argv[])
{
    return lr_classify(argc, argv);
}