diff --git a/src/main.rs b/src/main.rs index cd28d0e686bc6b83015fb2738fef9da4eb0bfa05..11958e191e8fb14d71a48c6b124baf2fb6987890 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 {