Hugging Face's logo

sakharamg
/
NMTKD

Model card Files Community
NMTKD
File size: 1,130 Bytes 
158b61b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
 
#include <cassert>
#include <iostream>
#include "SyntaxTree.h"
#include "Parameter.h"

using namespace std;

void SyntaxTree::Add(int startPos, int endPos, const std::string &label, const Parameter &params)
{
  //cerr << "add " << label << " to " << "[" << startPos << "-" << endPos << "]" << endl;

  Range range(startPos, endPos);
  Labels &labels = m_coll[range];

  bool add = true;
  if (labels.size()) {
    if (params.multiLabel == 1) {
      // delete the label in collection and add new
      assert(labels.size() == 1);
      labels.clear();
    } else if (params.multiLabel == 2) {
      // ignore this label
      add = false;
    }
  }

  if (add) {
    labels.push_back(label);
  }
}

void SyntaxTree::AddToAll(const std::string &label)
{
  Coll::iterator iter;
  for (iter = m_coll.begin(); iter != m_coll.end(); ++iter) {
    Labels &labels = iter->second;
    labels.push_back(label);
  }
}

const SyntaxTree::Labels &SyntaxTree::Find(int startPos, int endPos) const
{
  Coll::const_iterator iter;
  iter = m_coll.find(Range(startPos, endPos));
  return (iter == m_coll.end()) ? m_defaultLabels : iter->second;
}