set default context width

This commit is contained in:
svlandeg 2019-07-15 12:03:09 +02:00
parent 6e809e9b8b
commit 60f299374f

View File

@ -1093,6 +1093,8 @@ class EntityLinker(Pipe):
self.kb = None
self.cfg = dict(cfg)
self.sgd_context = None
if not self.cfg.get("context_width"):
self.cfg["context_width"] = 128
def set_kb(self, kb):
self.kb = kb