Examples with CompoundWord com.hankcs.hanlp.corpus.document.sentence.word.CompoundWord used on opensource projects

Example 1 with CompoundWord

use of com.hankcs.hanlp.corpus.document.sentence.word.CompoundWord in project HanLP by hankcs.

the class Document method getSimpleWordList.

public List<Word> getSimpleWordList() {
    List<IWord> wordList = getWordList();
    List<Word> simpleWordList = new LinkedList<Word>();
    for (IWord word : wordList) {
        if (word instanceof CompoundWord) {
            simpleWordList.addAll(((CompoundWord) word).innerList);
        } else {
            simpleWordList.add((Word) word);
        }
    }
    return simpleWordList;
}

Example 2 with CompoundWord

use of com.hankcs.hanlp.corpus.document.sentence.word.CompoundWord in project HanLP by hankcs.

the class Document method getSimpleSentenceList.

/**
     * 获取简单的句子列表，其中复合词会被拆分为简单词
     * @return
     */
public List<List<Word>> getSimpleSentenceList() {
    List<List<Word>> simpleList = new LinkedList<List<Word>>();
    for (Sentence sentence : sentenceList) {
        List<Word> wordList = new LinkedList<Word>();
        for (IWord word : sentence.wordList) {
            if (word instanceof CompoundWord) {
                for (Word inner : ((CompoundWord) word).innerList) {
                    wordList.add(inner);
                }
            } else {
                wordList.add((Word) word);
            }
        }
        simpleList.add(wordList);
    }
    return simpleList;
}

Example 3 with CompoundWord

use of com.hankcs.hanlp.corpus.document.sentence.word.CompoundWord in project HanLP by hankcs.

the class Document method getSimpleSentenceList.

/**
     * 获取简单的句子列表
     * @param spilt 如果为真，其中复合词会被拆分为简单词
     * @return
     */
public List<List<Word>> getSimpleSentenceList(boolean spilt) {
    List<List<Word>> simpleList = new LinkedList<List<Word>>();
    for (Sentence sentence : sentenceList) {
        List<Word> wordList = new LinkedList<Word>();
        for (IWord word : sentence.wordList) {
            if (word instanceof CompoundWord) {
                if (spilt) {
                    for (Word inner : ((CompoundWord) word).innerList) {
                        wordList.add(inner);
                    }
                } else {
                    wordList.add(((CompoundWord) word).toWord());
                }
            } else {
                wordList.add((Word) word);
            }
        }
        simpleList.add(wordList);
    }
    return simpleList;
}

Example 4 with CompoundWord

use of com.hankcs.hanlp.corpus.document.sentence.word.CompoundWord in project HanLP by hankcs.

the class Document method getSimpleSentenceList.

/**
     * 获取简单的句子列表，其中复合词的标签如果是set中指定的话会被拆分为简单词
     * @param labelSet
     * @return
     */
public List<List<Word>> getSimpleSentenceList(Set<String> labelSet) {
    List<List<Word>> simpleList = new LinkedList<List<Word>>();
    for (Sentence sentence : sentenceList) {
        List<Word> wordList = new LinkedList<Word>();
        for (IWord word : sentence.wordList) {
            if (word instanceof CompoundWord) {
                if (labelSet.contains(word.getLabel())) {
                    for (Word inner : ((CompoundWord) word).innerList) {
                        wordList.add(inner);
                    }
                } else {
                    wordList.add(((CompoundWord) word).toWord());
                }
            } else {
                wordList.add((Word) word);
            }
        }
        simpleList.add(wordList);
    }
    return simpleList;
}

Example 5 with CompoundWord

use of com.hankcs.hanlp.corpus.document.sentence.word.CompoundWord in project HanLP by hankcs.

the class NTDictionaryMaker method roleTag.

@Override
protected void roleTag(List<List<IWord>> sentenceList) {
    int i = 0;
    for (List<IWord> wordList : sentenceList) {
        Precompiler.compileWithoutNT(wordList);
        if (verbose) {
            System.out.print(++i + " / " + sentenceList.size() + " ");
            System.out.println("原始语料 " + wordList);
        }
        LinkedList<IWord> wordLinkedList = (LinkedList<IWord>) wordList;
        wordLinkedList.addFirst(new Word(Predefine.TAG_BIGIN, "S"));
        wordLinkedList.addLast(new Word(Predefine.TAG_END, "Z"));
        if (verbose)
            System.out.println("添加首尾 " + wordList);
        // 标注上文
        Iterator<IWord> iterator = wordLinkedList.iterator();
        IWord pre = iterator.next();
        while (iterator.hasNext()) {
            IWord current = iterator.next();
            if (current.getLabel().startsWith("nt") && !pre.getLabel().startsWith("nt")) {
                pre.setLabel(NT.A.toString());
            }
            pre = current;
        }
        if (verbose)
            System.out.println("标注上文 " + wordList);
        // 标注下文
        iterator = wordLinkedList.descendingIterator();
        pre = iterator.next();
        while (iterator.hasNext()) {
            IWord current = iterator.next();
            if (current.getLabel().startsWith("nt") && !pre.getLabel().startsWith("nt")) {
                pre.setLabel(NT.B.toString());
            }
            pre = current;
        }
        if (verbose)
            System.out.println("标注下文 " + wordList);
        // 标注中间
        {
            iterator = wordLinkedList.iterator();
            IWord first = iterator.next();
            IWord second = iterator.next();
            while (iterator.hasNext()) {
                IWord third = iterator.next();
                if (first.getLabel().startsWith("nt") && third.getLabel().startsWith("nt") && !second.getLabel().startsWith("nt")) {
                    second.setLabel(NT.X.toString());
                }
                first = second;
                second = third;
            }
            if (verbose)
                System.out.println("标注中间 " + wordList);
        }
        // 处理整个
        ListIterator<IWord> listIterator = wordLinkedList.listIterator();
        while (listIterator.hasNext()) {
            IWord word = listIterator.next();
            String label = word.getLabel();
            if (label.equals(label.toUpperCase()))
                continue;
            if (label.startsWith("nt")) {
                StringBuilder sbPattern = new StringBuilder();
                // 复杂机构
                if (word instanceof CompoundWord) {
                    listIterator.remove();
                    Word last = null;
                    for (Word inner : ((CompoundWord) word).innerList) {
                        last = inner;
                        String innerLabel = inner.label;
                        if (innerLabel.startsWith("ns")) {
                            inner.setValue(Predefine.TAG_PLACE);
                            inner.setLabel(NT.G.toString());
                            listIterator.add(inner);
                            sbPattern.append(inner.label);
                        } else if (innerLabel.startsWith("nt")) {
                            inner.value = Predefine.TAG_GROUP;
                            inner.label = NT.K.toString();
                            listIterator.add(inner);
                            sbPattern.append(inner.label);
                        } else if (innerLabel.equals("b") || innerLabel.equals("ng") || innerLabel.equals("j")) {
                            inner.label = NT.J.toString();
                            listIterator.add(inner);
                            sbPattern.append(inner.label);
                        } else if ("n".equals(innerLabel) || "an".equals(innerLabel) || "a".equals(innerLabel) || "vn".equals(innerLabel) || "vd".equals(innerLabel) || "vl".equals(innerLabel) || "v".equals(innerLabel) || "vi".equals(innerLabel) || "nnt".equals(innerLabel) || "nnd".equals(innerLabel) || "nf".equals(innerLabel) || "cc".equals(innerLabel) || "t".equals(innerLabel) || "z".equals(innerLabel)) {
                            inner.label = NT.C.toString();
                            listIterator.add(inner);
                            sbPattern.append(inner.label);
                        } else if ("nz".equals(innerLabel)) {
                            inner.label = NT.I.toString();
                            listIterator.add(inner);
                            sbPattern.append(inner.label);
                        } else if ("m".equals(innerLabel)) {
                            inner.value = Predefine.TAG_NUMBER;
                            inner.label = NT.M.toString();
                            listIterator.add(inner);
                            sbPattern.append(inner.label);
                        } else if ("w".equals(innerLabel)) {
                            inner.label = NT.W.toString();
                            listIterator.add(inner);
                            sbPattern.append(inner.label);
                        } else if (innerLabel.startsWith("nr") || "x".equals(innerLabel) || "nx".equals(innerLabel)) {
                            inner.value = Predefine.TAG_PEOPLE;
                            inner.label = NT.F.toString();
                            listIterator.add(inner);
                            sbPattern.append(inner.label);
                        } else if (innerLabel.startsWith("ni")) {
                            inner.label = NT.D.toString();
                            listIterator.add(inner);
                            sbPattern.append(inner.label);
                        } else if ("f".equals(innerLabel) || "s".equals(innerLabel)) {
                            inner.label = NT.L.toString();
                            listIterator.add(inner);
                            sbPattern.append(inner.label);
                        } else {
                            inner.label = NT.P.toString();
                            listIterator.add(inner);
                            sbPattern.append(inner.label);
                        }
                    }
                    if (last != null) {
                        last.label = NT.D.toString();
                        sbPattern.deleteCharAt(sbPattern.length() - 1);
                        sbPattern.append(last.label);
                        tfDictionary.add(sbPattern.toString());
                        sbPattern.setLength(0);
                    }
                } else {
                    word.setLabel(NT.K.toString());
                }
            } else {
                word.setLabel(NT.Z.toString());
            }
        }
        if (verbose)
            System.out.println("处理整个 " + wordList);
        wordLinkedList.getFirst().setLabel(NT.S.toString());
    }
}