From c799479b78608d7dc466e3cf69a145b094c4f508 Mon Sep 17 00:00:00 2001 From: Anders Olsson Date: Mon, 24 Feb 2020 21:49:54 +0100 Subject: [PATCH] Add method to check if item already has been added. --- src/model.rs | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/model.rs b/src/model.rs index 29e1bd3..c4cee97 100644 --- a/src/model.rs +++ b/src/model.rs @@ -47,6 +47,10 @@ impl BinaryModel { ); } + pub fn contains_item(&self, name: &str) -> bool { + self.storage.contains_key(name) + } + pub fn item_score(&self, name: &str, t: f64) -> (f64, f64) { let id = self.storage.get_id(name); let (ms, vs) = self.storage.item(id).fitter.predict(&[t]);