Candle CRF ports pytorch-crf to Huggingface Candle.
-
Pytorch CRF
crf = CRF(num_tags, batch_first=false)
-
Candle CRF
let crf = CRF::new(num_tags, false, &candle_core::Device::Cpu).unwrap();
-
Pytorch CRF
llh = crf(emissions, tags, mask)
-
Candle CRF
let llh = crf .forward(&emissions, &tags, Some(&mask), Reduction::default()) .unwrap();
-
Pytorch CRF
best_tags = crf.decode(emissions, mask)
-
Candle CRF
let best_tags = crf.decode(&emissions, Some(&mask)).unwrap();