mirror of
https://github.com/explosion/spaCy.git
synced 2025-01-26 09:14:32 +03:00
26 lines
571 B
Python
26 lines
571 B
Python
# coding: utf8
|
|
from __future__ import unicode_literals
|
|
|
|
import pytest
|
|
from spacy.tokens import Span
|
|
from spacy.language import Language
|
|
from spacy.pipeline.spancat import SpanCategorizer
|
|
|
|
|
|
def test_init_spancat(en_vocab):
|
|
model = SpanCategorizer(en_vocab)
|
|
|
|
|
|
@pytest.fixture
|
|
def spancat(en_vocab):
|
|
return SpanCategorizer(en_vocab)
|
|
|
|
|
|
def test_spancat_add_label(spancat):
|
|
assert "NP" not in spancat.labels
|
|
spancat.add_label("NP")
|
|
assert "NP" in spancat.labels
|
|
|
|
def test_spancat_init_model(spancat):
|
|
model = spancat.Model(10, width=10, embed_size=10)
|