Update to askalono 0.4.0 and use a more strict confidence threshold

o2sh · Nov 9, 2019 · 6ae318c · 6ae318c
1 parent f74f741
commit 6ae318c
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 6 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -12,7 +12,7 @@ exclude = ["assets/*.png"]
 colored= "1.8.0"
 git2 = {version = "0.7.5", default-features = false}
 tokei = "10.0"
-askalono = { git = "https://github.com/amzn/askalono.git", rev = "39f6e228775e17266bcba46f48b26b01bbfb7698" }
+askalono = "0.4.0"
 bytecount = "0.5.1"
 clap = "2.33.0"
 strum = "0.16.0"

diff --git a/src/license.rs b/src/license.rs
@@ -5,6 +5,7 @@ use crate::Error;
 type Result<T> = std::result::Result<T, Error>;
 
 static CACHE_DATA: &[u8] = include_bytes!("../resources/licenses/cache.bin.zstd");
+const MIN_THRESHOLD: f32 = 0.8;
 
 pub struct Detector {
     store: Store,
@@ -20,7 +21,7 @@ impl Detector {
     pub fn analyze(&self, text: &str) -> Option<String> {
         let matched = self.store.analyze(&TextData::from(text));
 
-        if matched.score > 0. {
+        if matched.score >= MIN_THRESHOLD {
             Some(matched.name.into())
         } else {
             None