diff options
Diffstat (limited to 'rst_parser/arc_ff.cc')
| -rw-r--r-- | rst_parser/arc_ff.cc | 18 | 
1 files changed, 10 insertions, 8 deletions
diff --git a/rst_parser/arc_ff.cc b/rst_parser/arc_ff.cc index 2992cde2..b3560b73 100644 --- a/rst_parser/arc_ff.cc +++ b/rst_parser/arc_ff.cc @@ -38,28 +38,28 @@ struct ArcFFImpl {    template <typename A, typename B>    static void Fire(SparseVector<weight_t>* v, const A& a, const B& b) {      ostringstream os; -    os << a << '_' << b; +    os << a << ':' << b;      v->set_value(FD::Convert(os.str()), 1);    }    template <typename A, typename B, typename C>    static void Fire(SparseVector<weight_t>* v, const A& a, const B& b, const C& c) {      ostringstream os; -    os << a << '_' << b << '_' << c; +    os << a << ':' << b << '_' << c;      v->set_value(FD::Convert(os.str()), 1);    }    template <typename A, typename B, typename C, typename D>    static void Fire(SparseVector<weight_t>* v, const A& a, const B& b, const C& c, const D& d) {      ostringstream os; -    os << a << '_' << b << '_' << c << '_' << d; +    os << a << ':' << b << '_' << c << '_' << d;      v->set_value(FD::Convert(os.str()), 1);    }    template <typename A, typename B, typename C, typename D, typename E>    static void Fire(SparseVector<weight_t>* v, const A& a, const B& b, const C& c, const D& d, const E& e) {      ostringstream os; -    os << a << '_' << b << '_' << c << '_' << d << '_' << e; +    os << a << ':' << b << '_' << c << '_' << d << '_' << e;      v->set_value(FD::Convert(os.str()), 1);    } @@ -75,7 +75,7 @@ struct ArcFFImpl {      const string& mod_pos = TD::Convert(sent.pos[m]);      const string& mod_pos_L = (m > 0 ? TD::Convert(sent.pos[m-1]) : kLEFT_POS);      const string& mod_pos_R = (m < sent.pos.size() - 1 ? TD::Convert(sent.pos[m]) : kRIGHT_POS); -    const bool dir = m < h; +    const string dir = (m < h ? "MLeft" : "MRight");      int v = m - h;      if (v < 0) {        v= -1 - int(log(-v) / log(2)); @@ -118,11 +118,13 @@ struct ArcFFImpl {        // between features        int left = min(h,m);        int right = max(h,m); -      if (right - left > 2) { -        ++left; +      if (right - left >= 2) { +        --right;          for (map<WordID, vector<int> >::const_iterator it = pcs.begin(); it != pcs.end(); ++it) { -          if (it->second[left] != it->second[right]) +          if (it->second[left] != it->second[right]) {              Fire(features, "BT", head_pos, TD::Convert(it->first), mod_pos); +            Fire(features, "BT", head_pos, TD::Convert(it->first), mod_pos, dir); +          }          }        }      }  | 
