Получить тип дерева NLTK () - PullRequest
Новая
       31

Получить тип дерева NLTK ()

0 голосов
/ 19 июня 2020

Я использую модуль Semcor NLTK: 

nltk.download('semcor')
from nltk.corpus import semcor

semcor.tagged_sents() выполняет итерацию по тем же предложениям с дополнительной аннотацией, включая идентификаторы леммы Wor dNet. 

semcor.tagged_sents(tag="sem")[0]
>>> [['The'],
 Tree(Lemma('group.n.01.group'), [Tree('NE', ['Fulton', 'County', 'Grand', 'Jury'])]),
 Tree(Lemma('state.v.01.say'), ['said']),
 Tree(Lemma('friday.n.01.Friday'), ['Friday']),
 ['an'],
 Tree(Lemma('probe.n.01.investigation'), ['investigation']),
 ['of'],
 Tree(Lemma('atlanta.n.01.Atlanta'), ['Atlanta']),
 ["'s"],
 Tree(Lemma('late.s.03.recent'), ['recent']),
 Tree(Lemma('primary.n.01.primary_election'), ['primary', 'election']),
 Tree(Lemma('produce.v.04.produce'), ['produced']),
 ['``'],
 ['no'],
 Tree(Lemma('evidence.n.01.evidence'), ['evidence']),
 ["''"],
 ['that'],
 ['any'],
 Tree(Lemma('abnormality.n.04.irregularity'), ['irregularities']),
 Tree(Lemma('happen.v.01.take_place'), ['took', 'place']),
 ['.']]

Когда я использую индексов в этом списке, я получаю следующий результат: 

semcor.tagged_sents(tag="sem")[0][1][0]
>>> Tree('NE', ['Fulton', 'County', 'Grand', 'Jury'])

Когда я использую еще один индекс, я получаю токены из списка в качестве выходных: 

semcor.tagged_sents(tag="sem")[0][1][0][0]
>>> 'Fulton'

Моя цель - два -fold:

Какой код я могу использовать, чтобы получить лемму в качестве вывода? Таким образом, вывод будет: 

>>> Tree(Lemma('group.n.01.group')

И какой код я могу использовать, чтобы получить тип дерева в качестве вывода? В случае этого примера: 

>>> 'NE'

1 Ответ

1 голос
/ 19 июня 2020 
semcor.tagged_sents(tag="sem")[0][1].label()
#'group.n.01'

semcor.tagged_sents(tag="sem")[0][1][0].label()
#'NE'
...