A resposta por https://stackoverflow.com/users/1515832/generic-human é ótima. Mas a melhor implementação disso que já vi foi escrita pelo próprio Peter Norvig em seu livro 'Beautiful Data'.
Antes de colar seu código, deixe-me explicar por que o método de Norvig é mais preciso (embora um pouco mais lento e longo em termos de código).
1) Os dados são um pouco melhores - tanto em termos de tamanho quanto em termos de precisão (ele usa uma contagem de palavras em vez de uma classificação simples) 2) Mais importante, é a lógica por trás de n-gramas que realmente torna a abordagem tão precisa .
O exemplo que ele fornece em seu livro é o problema de dividir uma string 'sitdown'. Agora, um método não bigrama de divisão de string consideraria p ('sentar') * p ('para baixo'), e se for menor que p ('sentar') - o que será o caso com bastante frequência - NÃO irá dividir , mas gostaríamos que fizesse (na maioria das vezes).
No entanto, quando você tem o modelo de bigrama, você pode avaliar p ('sentar') como um bigrama versus p ('sentar') e o anterior ganha. Basicamente, se você não usar bigramas, ele trata a probabilidade das palavras que você está dividindo como independentes, o que não é o caso, algumas palavras têm mais probabilidade de aparecer uma após a outra. Infelizmente, essas também são as palavras que costumam ficar juntas em muitos casos e confundem o divisor.
Aqui está o link para os dados (são dados para 3 problemas separados e segmentação é apenas um. Leia o capítulo para obter detalhes): http://norvig.com/ngrams/
e aqui está o link para o código: http://norvig.com/ngrams/ngrams.py
Esses links já estão no ar há algum tempo, mas vou copiar e colar a parte de segmentação do código aqui mesmo assim
import re, string, random, glob, operator, heapq
from collections import defaultdict
from math import log10
def memo(f):
"Memoize function f."
table = {}
def fmemo(*args):
if args not in table:
table[args] = f(*args)
return table[args]
fmemo.memo = table
return fmemo
def test(verbose=None):
"""Run some tests, taken from the chapter.
Since the hillclimbing algorithm is randomized, some tests may fail."""
import doctest
print 'Running tests...'
doctest.testfile('ngrams-test.txt', verbose=verbose)
################ Word Segmentation (p. 223)
@memo
def segment(text):
"Return a list of words that is the best segmentation of text."
if not text: return []
candidates = ([first]+segment(rem) for first,rem in splits(text))
return max(candidates, key=Pwords)
def splits(text, L=20):
"Return a list of all possible (first, rem) pairs, len(first)<=L."
return [(text[:i+1], text[i+1:])
for i in range(min(len(text), L))]
def Pwords(words):
"The Naive Bayes probability of a sequence of words."
return product(Pw(w) for w in words)
#### Support functions (p. 224)
def product(nums):
"Return the product of a sequence of numbers."
return reduce(operator.mul, nums, 1)
class Pdist(dict):
"A probability distribution estimated from counts in datafile."
def __init__(self, data=[], N=None, missingfn=None):
for key,count in data:
self[key] = self.get(key, 0) + int(count)
self.N = float(N or sum(self.itervalues()))
self.missingfn = missingfn or (lambda k, N: 1./N)
def __call__(self, key):
if key in self: return self[key]/self.N
else: return self.missingfn(key, self.N)
def datafile(name, sep='\t'):
"Read key,value pairs from file."
for line in file(name):
yield line.split(sep)
def avoid_long_words(key, N):
"Estimate the probability of an unknown word."
return 10./(N * 10**len(key))
N = 1024908267229 ## Number of tokens
Pw = Pdist(datafile('count_1w.txt'), N, avoid_long_words)
#### segment2: second version, with bigram counts, (p. 226-227)
def cPw(word, prev):
"Conditional probability of word, given previous word."
try:
return P2w[prev + ' ' + word]/float(Pw[prev])
except KeyError:
return Pw(word)
P2w = Pdist(datafile('count_2w.txt'), N)
@memo
def segment2(text, prev='<S>'):
"Return (log P(words), words), where words is the best segmentation."
if not text: return 0.0, []
candidates = [combine(log10(cPw(first, prev)), first, segment2(rem, first))
for first,rem in splits(text)]
return max(candidates)
def combine(Pfirst, first, (Prem, rem)):
"Combine first and rem results into one (probability, words) pair."
return Pfirst+Prem, [first]+rem