def test_BIO_shape(BIO): assert BIO.shape == (len(IOBv), len(IOBv)) mask = transition_mask(IOBv, "IOB2", IOBv["<GO>"], IOBv["<EOS>"], IOBv["<PAD>"]) assert mask.shape == (len(IOBv), len(IOBv))
def BIO(): return transition_mask(IOBv, "BIO", IOBv["<GO>"], IOBv["<EOS>"], IOBv["<PAD>"])
def IOBES(): return transition_mask(IOBESv, "IOBES", IOBESv["<GO>"], IOBESv["<EOS>"], IOBESv["<PAD>"])
def IOB(): return transition_mask(IOBv, "IOB", IOBv["<GO>"], IOBv["<EOS>"], IOBv["<PAD>"])