diff options
author | Chris Dyer <cdyer@cs.cmu.edu> | 2012-04-18 22:37:08 -0400 |
---|---|---|
committer | Chris Dyer <cdyer@cs.cmu.edu> | 2012-04-18 22:37:08 -0400 |
commit | 9f76fbba192037994421513673014c97929fd5d2 (patch) | |
tree | 9055942a42dc42fae052e943c2a5d9b477e104b6 /rst_parser/dep_training.h | |
parent | f419e830faa6c47a8c0ba55fc0d10a113405a210 (diff) |
refactor
Diffstat (limited to 'rst_parser/dep_training.h')
-rw-r--r-- | rst_parser/dep_training.h | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/rst_parser/dep_training.h b/rst_parser/dep_training.h index 73ffd298..3eeee22e 100644 --- a/rst_parser/dep_training.h +++ b/rst_parser/dep_training.h @@ -1,6 +1,7 @@ #ifndef _DEP_TRAINING_H_ #define _DEP_TRAINING_H_ +#include <iostream> #include <string> #include <vector> #include "arc_factored.h" @@ -11,7 +12,8 @@ struct TrainingInstance { EdgeSubset tree; SparseVector<weight_t> features; // reads a "Jsent" formatted dependency file - static void ReadTraining(const std::string& fname, std::vector<TrainingInstance>* corpus, int rank = 0, int size = 1); + static bool ReadInstance(std::istream* in, TrainingInstance* instance); // returns false at EOF + static void ReadTrainingCorpus(const std::string& fname, std::vector<TrainingInstance>* corpus, int rank = 0, int size = 1); }; #endif |