parse_regressor.h


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

/*
Copyright (c) 2009 Yahoo! Inc.  All rights reserved.  The copyrights
embodied in the content of this file are licensed under the BSD
(revised) open source license
 */

#ifndef PR_H
#define PR_H
#include "loss_functions.h"
#include "boost/program_options.hpp"

namespace po = boost::program_options;

typedef float weight;

struct regressor {
  weight** weight_vectors;
  weight** regularizers;

  loss_function *loss;
};

void parse_regressor_args(po::variables_map& vm, regressor& r, string& final_regressor_name, bool quiet);

void initialize_regressor(regressor &r);

void finalize_regressor(string reg_name, regressor &r);
void dump_regressor(string reg_name, regressor &r, bool as_text=0, bool reg_vector=0);

#endif