-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathOptions.h
82 lines (60 loc) · 1.31 KB
/
Options.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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
/*
* Options.h
*
* Created on: Mar 27, 2014
* Author: yuanz
*/
#ifndef OPTIONS_H_
#define OPTIONS_H_
#include <string>
namespace segparser {
using namespace std;
class Options {
public:
public:
// file name
string trainFile;
string testFile;
string outFile;
string modelName;
int lang;
// model type
bool train;
bool test;
bool trainPruner;
int learningMode;
int testingMode;
// parameter
int numIters;
int maxHead;
double pruneThresh;
int trainSentences;
int testSentences;
int maxLength; // maximum length of the sentences during *training*
int devThread;
int trainThread; // only useful when hill climbing training
int seed;
double regC;
// feature;
bool useCS; // consecutive sibling
bool useGP; // grandparent
bool useHO; // grand-sibling, tri-sibling and high order and global
bool useSP; // seg pos feature
int trainConvergeIter; // for hill climbing
int testConvergeIter;
bool evalPunc;
bool useTedEval;
bool jointSegPos; // joint model or pipeline
int earlyStop; // early stop strategy in training
bool saveBestModel;
double bestScore;
Options();
virtual ~Options();
void processArguments(int argc, char** argv);
void setPrunerOptions();
void outputArg();
private:
int findLang(string file);
};
} /* namespace segparser */
#endif /* OPTIONS_H_ */