Python transition_mask Examples

Example #1

File: test_transition_masks.py Project: dpressel/mead-layers

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))

Example #2

File: test_transition_masks.py Project: dpressel/mead-layers

def BIO():
    return transition_mask(IOBv, "BIO", IOBv["<GO>"], IOBv["<EOS>"],
                           IOBv["<PAD>"])

Example #3

File: test_transition_masks.py Project: dpressel/mead-layers

def IOBES():
    return transition_mask(IOBESv, "IOBES", IOBESv["<GO>"], IOBESv["<EOS>"],
                           IOBESv["<PAD>"])

Example #4

File: test_transition_masks.py Project: dpressel/mead-layers

def IOB():
    return transition_mask(IOBv, "IOB", IOBv["<GO>"], IOBv["<EOS>"],
                           IOBv["<PAD>"])