蜡烛 CRF

蜡烛 CRF 将 pytorch-crf 迁移到 Huggingface Candle。

初始化

• Pytorch CRF

  crf = CRF(num_tags, batch_first=false)
  

• 蜡烛 CRF

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

正向传播

• Pytorch CRF

  llh = crf(emissions, tags, mask)
  

• 蜡烛 CRF

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

解码

• Pytorch CRF

  best_tags = crf.decode(emissions, mask)
  

• 蜡烛 CRF

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