instant-labs
/
instant-segment
mirror of https://github.com/instant-labs/instant-segment.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Simplify bigram scoring algorithm

This commit is contained in:
Dirkjan Ochtman 2020-12-07 14:24:33 +01:00
parent f26793379b
commit c571996925
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/lib.rs
View File

@ -42,12 +42,12 @@ impl Segmenter {
fn score(&self, word: &str, previous: Option<&str>) -> f64 {
if let Some(prev) = previous {
if let Some(pb) = self.bigrams.get(&(prev.into(), word.into())) {
if self.unigrams.get(prev).is_some() {
if let Some(bi) = self.bigrams.get(&(prev.into(), word.into())) {
if let Some(uni) = self.unigrams.get(prev) {
// Conditional probability of the word given the previous
// word. The technical name is "stupid backoff" and it's
// not a probability distribution but it works well in practice.
return pb / self.total / self.score(prev, None);
return (bi / self.total) / (uni / self.total);
}
}
}