langml.layers.crf

Module Contents

Classes

CRF

class langml.layers.crf.CRF(output_dim: int, sparse_target: bool = True, **kwargs)[source]

Bases: tensorflow.keras.layers.Layer

build(self, input_shape: langml.tensor_typing.Tensors)[source]
compute_mask(self, inputs: langml.tensor_typing.Tensors, mask: Optional[langml.tensor_typing.Tensors] = None)[source]
call(self, inputs: langml.tensor_typing.Tensors, sequence_lengths: Optional[langml.tensor_typing.Tensors] = None, training: Optional[Union[bool, int]] = None, mask: Optional[langml.tensor_typing.Tensors] = None, **kwargs) → langml.tensor_typing.Tensors[source]
property loss(self) → Callable[source]
property accuracy(self) → Callable[source]
compute_output_shape(self, input_shape: langml.tensor_typing.Tensors) → langml.tensor_typing.Tensors[source]
property trans(self) → langml.tensor_typing.Tensors[source]

transition parameters

get_config(self) → dict[source]
static get_custom_objects() → dict[source]