explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-09-21 11:29:13 +03:00
Code Issues Packages Projects Releases Wiki Activity

* Switch between the orth and char codecs depending on which is shorter for that message. Mostly orth is shorter, except if there are OOV words.

Browse Source
This commit is contained in:
Matthew Honnibal 2015-07-20 01:36:22 +02:00
parent 5a042ee0d3
commit 5a7d060d9c
1 changed files with 26 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
spacy/serialize/packer.pyx
View File

@ -1,7 +1,7 @@
# cython: profile=True
from __future__ import unicode_literals
from libc.stdint cimport uint32_t
from libc.stdint cimport uint32_t, int32_t
from libc.stdint cimport uint64_t
from libc.math cimport exp as c_exp
from libcpp.queue cimport priority_queue
@ -68,7 +68,7 @@ def _gen_orths(Vocab vocab):
def _gen_chars(Vocab vocab):
cdef attr_t orth
cdef size_t addr
char_weights = {u' ': 0.0}
char_weights = {chr(i): 1e-20 for i in range(256)}
cdef unicode string
cdef bytes char
cdef bytes utf8_str
@ -79,15 +79,17 @@ def _gen_chars(Vocab vocab):
for char in utf8_str:
char_weights.setdefault(char, 0.0)
char_weights[char] += c_exp(lex.prob)
char_weights[u' '] += c_exp(lex.prob)
char_weights[b' '] += c_exp(lex.prob)
return char_weights.items()
cdef class Packer:
def __init__(self, Vocab vocab, attr_freqs):
def __init__(self, Vocab vocab, attr_freqs, char_freqs=None):
if char_freqs is None:
char_freqs = _gen_chars(vocab)
self.vocab = vocab
self.orth_codec = HuffmanCodec(_gen_orths(vocab))
self.char_codec = HuffmanCodec(_gen_chars(vocab))
self.char_codec = HuffmanCodec(char_freqs)
codecs = []
attrs = []
@ -104,7 +106,14 @@ cdef class Packer:
return cls(vocab, util.read_encoding_freqs(data_dir))
def pack(self, Doc doc):
bits = self._orth_encode(doc)
orths = [t.orth for t in doc]
chars = doc.string.encode('utf8')
# n_bits returns nan for oov words, i.e. can't encode message.
# So, it's important to write the conditional like this.
if self.orth_codec.n_bits(orths) < self.char_codec.n_bits(chars, overhead=1):
bits = self._orth_encode(doc)
else:
bits = self._char_encode(doc)
array = doc.to_array(self.attrs)
for i, codec in enumerate(self._codecs):
codec.encode(array[:, i], bits)
@ -112,8 +121,11 @@ cdef class Packer:
def unpack(self, BitArray bits):
bits.seek(0)
cdef uint32_t length = bits.read32()
doc = self._orth_decode(bits, length)
cdef int32_t length = bits.read32()
if length >= 0:
doc = self._orth_decode(bits, length)
else:
doc = self._char_decode(bits, -length)
array = numpy.zeros(shape=(len(doc), len(self._codecs)), dtype=numpy.int32)
for i, codec in enumerate(self._codecs):
@ -125,8 +137,8 @@ cdef class Packer:
def _orth_encode(self, Doc doc):
cdef BitArray bits = BitArray()
orths = [w.orth for w in doc]
cdef uint32_t length = len(doc)
bits.extend(length, 32)
cdef int32_t length = len(doc)
bits.extend(length, 32)
self.orth_codec.encode(orths, bits)
for token in doc:
bits.append(bool(token.whitespace_))
@ -142,11 +154,10 @@ cdef class Packer:
def _char_encode(self, Doc doc):
cdef BitArray bits = BitArray()
cdef bytes utf8_str = doc.string.encode('utf8')
cdef uint32_t length = len(utf8_str)
bits.extend(length, 32)
cdef bytes utf8_string = doc.string.encode('utf8')
self.char_codec.encode(utf8_string, bits)
cdef int32_t length = len(utf8_str)
# Signal chars with negative length
bits.extend(-length, 32)
self.char_codec.encode(utf8_str, bits)
for token in doc:
for i in range(len(token)-1):
bits.append(False)