Skip to content
Snippets Groups Projects
Commit e4abccd7 authored by Phil Höfer's avatar Phil Höfer
Browse files

Implement Config Option for Term Frequency Threshold

parent 8d92618b
No related branches found
No related tags found
1 merge request!4Resolve "Implement Reading of Configuration from File"
...@@ -26,7 +26,13 @@ fn main() -> Result<(), io::Error> { ...@@ -26,7 +26,13 @@ fn main() -> Result<(), io::Error> {
let markov_chain = build_markov_chain("../../data/data.csv") let markov_chain = build_markov_chain("../../data/data.csv")
.unwrap_or(build_markov_chain("data.csv") .unwrap_or(build_markov_chain("data.csv")
.unwrap_or_default()); .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 // Print the Markov Chain for verification
for (key, values) in &markov_chain { for (key, values) in &markov_chain {
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment