Augustus 3.4.0
Loading...
Searching...
No Matches
train_logReg_param.hh
1/*
2 * train_logReg_param.hh
3 *
4 * License: Artistic License, see file LICENSE.TXT or
5 * https://opensource.org/licenses/artistic-license-1.0
6 */
7
8#ifndef _TRAINLOGREG_HH
9#define _TRAINLOGREG_HH
10
11// project includes
12
13#include "types.hh"
14#include "properties.hh"
15
16// standard C/C++ includes
17#include <iostream>
18#include <vector>
19#include <unordered_map>
20
21#include <gsl/gsl_multimin.h>
22
23using namespace std;
24
32
33public:
34 train_data(unordered_map<string, pair<int, vector<double> > > *samples, unordered_map<string,int> *ref_class, int numSp);
35 vector<pair<int, vector<double> > > *exon_samples;
36 vector<pair<int, vector<double> > > *intron_samples;
37 int num_features;
38 double prob_true_false_exons;
39 double prob_true_false_introns;
40 vector<int> exon_feature_idx;
41 vector<int> intron_feature_idx;
42 void set_mean_std(vector<pair<int, vector<double> > > *samples, vector<int> feature_idx);
43
44};
45
46double activation_f(const gsl_vector *theta, vector<double> *f);
47double get_feature(vector<double> *f, int i);
48double cross_entropy_error_f(const gsl_vector *theta, void *features);
49void cross_entropy_error_df(const gsl_vector *theta, void *features, gsl_vector *df);
50void cross_entropy_error_fdf(const gsl_vector *params, void *features, double *f, gsl_vector *df);
51double rob_cross_entropy_error_f(const gsl_vector *theta, void *features);
52void rob_cross_entropy_error_df(const gsl_vector *theta, void *features, gsl_vector *df);
53void rob_cross_entropy_error_fdf(const gsl_vector *params, void *features, double *f, gsl_vector *df);
54void optimize_parameters(train_data *data);
55void reference_from_file(unordered_map<string,int> *ref_class);
56void trainFeature_from_file();
57void train_OEscore_params(int numSpecies);
58void sgd(vector<pair<int, vector<double> > > *samples, gsl_vector *theta, int n);
59void gsl_minimizer(vector<pair<int, vector<double> > > *samples, gsl_vector *theta, int n);
60
61#endif
implementation of logistic regression for training OE and EC feature parameters using GSL
Definition train_logReg_param.hh:31