Skip to main content

Module structured

Module structured 

Source
Expand description

Differentiable structured-prediction layers.

Currently the sinkhorn_crf module: entropy-regularised optimal-transport (Sinkhorn) normalisation as a differentiable replacement for sum-product normalisation in matching / permutation structured prediction.

Re-exports§

pub use sinkhorn_crf::SinkhornConfig;
pub use sinkhorn_crf::SinkhornCrf;
pub use sinkhorn_crf::SinkhornResult;
pub use sinkhorn_crf::sinkhorn_normalize;
pub use sinkhorn_crf::sinkhorn_normalize_with_margins;

Modules§

sinkhorn_crf
Sinkhorn CRF: entropy-regularised optimal-transport normalisation for structured prediction.