Как создать окно / чанк для списка предложений?
У меня есть список предложений, и я хочу создать скипграмму (window size = 3)
но я не хочу, чтобы счетчик охватывал предложения, поскольку все они не связаны между собой.
Итак, если у меня есть предложения:
[["my name is John"] , ["This PC is black"]]
триплеты будут:
[my name is]
[name is john]
[this PC is]
[PC is black]
Каков наилучший способ сделать это?
3 ответа
Решение
Попробуй это!
from nltk import ngrams
def generate_ngrams(sentences,window_size =3):
for sentence in sentences:
yield from ngrams(sentence[0].split(), window_size)
sentences= [["my name is John"] , ["This PC is black"]]
for c in generate_ngrams(sentences,3):
print (c)
#output:
('my', 'name', 'is')
('name', 'is', 'John')
('This', 'PC', 'is')
('PC', 'is', 'black')
Вот простая функция, чтобы сделать это.
def skipgram(corpus, window_size = 3):
sg = []
for sent in corpus:
sent = sent[0].split()
if len(sent) <= window_size:
sg.append(sent)
else:
for i in range(0, len(sent)-window_size+1):
sg.append(sent[i: i+window_size])
return sg
corpus = [["my name is John"] , ["This PC is black"]]
skipgram(corups)
Вы действительно не хотите skipgram
как таковой, но вы хотите кусок по размеру, попробуйте это:
from lazyme import per_chunk
tokens = "my name is John".split()
list(per_chunk(tokens, 2))
[из]:
[('my', 'name'), ('is', 'John')]
Если вы хотите скользящее окно, т.е. ngrams
:
from lazyme import per_window
tokens = "my name is John".split()
list(per_window(tokens, 2))
[из]:
[('my', 'name'), ('name', 'is'), ('is', 'John')]
Аналогично в НЛТК для нграмм:
from nltk import ngrams
tokens = "my name is John".split()
list(ngrams(tokens, 2))
[из]:
[('my', 'name'), ('name', 'is'), ('is', 'John')]
Если вы хотите фактические skipgrams, Как вычислить skipgram в Python?
from nltk import skipgrams
tokens = "my name is John".split()
list(skipgrams(tokens, n=2, k=1))
[из]:
[('my', 'name'),
('my', 'is'),
('name', 'is'),
('name', 'John'),
('is', 'John')]