Candle CRF

Candle CRF ports pytorch-crf to Huggingface Candle.

Initialization

• Pytorch CRF

  crf = CRF(num_tags, batch_first=false)

• Candle CRF

  let crf = CRF::new(num_tags, false, &candle_core::Device::Cpu).unwrap();

Forward

• Pytorch CRF

  llh = crf(emissions, tags, mask)

• Candle CRF

  let llh = crf
          .forward(&emissions, &tags, Some(&mask), Reduction::default())
          .unwrap();

Decode

• Pytorch CRF

  best_tags = crf.decode(emissions, mask)

• Candle CRF

  let best_tags = crf.decode(&emissions, Some(&mask)).unwrap();