From e4abccd7d6cf5c5da1200b59c67aa49e1ab44f68 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Phil=20H=C3=B6fer?= <phil.hoefer@suma-ev.de>
Date: Thu, 20 Jun 2024 14:47:25 +0200
Subject: [PATCH] Implement Config Option for Term Frequency Threshold
---
src/main.rs | 8 +++++++-
1 file changed, 7 insertions(+), 1 deletion(-)
diff --git a/src/main.rs b/src/main.rs
index cd28d0e..11958e1 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -26,7 +26,13 @@ fn main() -> Result<(), io::Error> {
let markov_chain = build_markov_chain("../../data/data.csv")
.unwrap_or(build_markov_chain("data.csv")
.unwrap_or_default());
- let filtered_markov_chain = filter_markov_chain(&markov_chain,2);
+
+
+ let term_frequency_threshold = match config.get("term_frequency_threshold") {
+ Some(toml::Value::Integer(n)) if *n >= 0 => *n as usize,
+ _ => 2
+ };
+ let filtered_markov_chain = filter_markov_chain(&markov_chain,term_frequency_threshold);
// Print the Markov Chain for verification
for (key, values) in &markov_chain {
--
GitLab