summaryrefslogtreecommitdiff
path: root/dtrain/common.h
diff options
context:
space:
mode:
Diffstat (limited to 'dtrain/common.h')
-rw-r--r--dtrain/common.h15
1 files changed, 10 insertions, 5 deletions
diff --git a/dtrain/common.h b/dtrain/common.h
index cf365d48..706f51c2 100644
--- a/dtrain/common.h
+++ b/dtrain/common.h
@@ -7,7 +7,9 @@
#include <vector>
#include <cassert>
#include <cmath>
+#include <iomanip>
+// cdec includes
#include "sentence_metadata.h"
#include "verbose.h"
#include "viterbi.h"
@@ -16,16 +18,19 @@
#include "decoder.h"
#include "weights.h"
+// boost includes
#include <boost/algorithm/string.hpp>
#include <boost/program_options.hpp>
+// own headers
#include "score.h"
-#define DTRAIN_DEFAULT_K 100
-#define DTRAIN_DEFAULT_N 4
-#define DTRAIN_DEFAULT_T 1
-
-#define DTRAIN_DOTOUT 100
+#define DTRAIN_DEFAULT_K 100 // k for kbest lists
+#define DTRAIN_DEFAULT_N 4 // N for ngrams (e.g. BLEU)
+#define DTRAIN_DEFAULT_T 1 // iterations
+#define DTRAIN_DEFAULT_SCORER "stupid_bleu" // scorer
+#define DTRAIN_DOTS 100 // when to display a '.'
+#define DTRAIN_TMP_DIR "/tmp" // put this on a SSD?
using namespace std;