From 0e9b99c2d1eb7e9b3cf94b56fc3bb1dae8f5491f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Phil=20H=C3=B6fer?= <phil.hoefer@suma-ev.de>
Date: Fri, 28 Jun 2024 15:55:29 +0200
Subject: [PATCH] Expand data file locations
---
src/main.rs | 5 +++--
src/predictors/basic_set.rs | 2 ++
2 files changed, 5 insertions(+), 2 deletions(-)
diff --git a/src/main.rs b/src/main.rs
index fbc979a..a72b58c 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -39,8 +39,9 @@ fn main() -> Result<(), io::Error> {
}
let markov_chain = basic_set::from_file_path_and_config("../../data/data.csv",config.clone())
- .unwrap_or(basic_set::from_file_path_and_config("data.csv",config.clone())
- .unwrap_or(basic_set::SetPredictor::new()));
+ .unwrap_or(basic_set::from_file_path_and_config("data/data.csv",config.clone())
+ .unwrap_or(basic_set::from_file_path_and_config("data.csv",config.clone())
+ .unwrap_or(basic_set::SetPredictor::new()));
// let term_frequency_threshold = match config.get("term_frequency_threshold") {
diff --git a/src/predictors/basic_set.rs b/src/predictors/basic_set.rs
index 56bc0be..ee00933 100644
--- a/src/predictors/basic_set.rs
+++ b/src/predictors/basic_set.rs
@@ -109,7 +109,9 @@ pub fn from_file_path_and_config(file_path: &str, config: HashMap<String, impl I
for (key, value) in config {
configuration.insert(key, value.into());
}
+ println!("Trying to open data file at {}",file_path);
let file = File::open(file_path)?;
+ println!("Reading data file...");
let mut reader = ReaderBuilder::new().from_reader(file);
let mut markov_chain: SetPredictor = SetPredictor::new();
markov_chain.configuration = configuration;
--
GitLab