From 54e109c7a68f71cc39cdcfde4a1c51a639b6da49 Mon Sep 17 00:00:00 2001
From: qnater <quentin.nater@gmail.com>
Date: Fri, 27 Sep 2024 17:30:12 +0200
Subject: [PATCH] 10. adding of eeg dataset with bayesian tests

---
 .idea/workspace.xml                          | 8 ++++++--
 tests/env/optimal_parameters_stmvl_test.toml | 4 ++++
 tests/test_opti_bayesian_cdrec.py            | 2 +-
 tests/test_opti_bayesian_iim.py              | 2 +-
 tests/test_opti_bayesian_mrnn.py             | 2 +-
 tests/test_opti_bayesian_stmvl.py            | 2 +-
 6 files changed, 14 insertions(+), 6 deletions(-)
 create mode 100644 tests/env/optimal_parameters_stmvl_test.toml

diff --git a/.idea/workspace.xml b/.idea/workspace.xml
index e7c2034..58f8aa3 100644
--- a/.idea/workspace.xml
+++ b/.idea/workspace.xml
@@ -2,7 +2,11 @@
 <project version="4">
   <component name="ChangeListManager">
     <list default="true" id="d2565aa5-b0c7-4b53-9b23-84bd42c9e5d7" name="Changes" comment="">
-      <change beforePath="$PROJECT_DIR$/imputegap/wrapper/AlgoPython/MRNN/Data_Loader.py" beforeDir="false" afterPath="$PROJECT_DIR$/imputegap/wrapper/AlgoPython/MRNN/Data_Loader.py" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/tests/test_opti_bayesian_cdrec.py" beforeDir="false" afterPath="$PROJECT_DIR$/tests/test_opti_bayesian_cdrec.py" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/tests/test_opti_bayesian_iim.py" beforeDir="false" afterPath="$PROJECT_DIR$/tests/test_opti_bayesian_iim.py" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/tests/test_opti_bayesian_mrnn.py" beforeDir="false" afterPath="$PROJECT_DIR$/tests/test_opti_bayesian_mrnn.py" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/tests/test_opti_bayesian_stmvl.py" beforeDir="false" afterPath="$PROJECT_DIR$/tests/test_opti_bayesian_stmvl.py" afterDir="false" />
     </list>
     <option name="SHOW_DIALOG" value="false" />
     <option name="HIGHLIGHT_CONFLICTS" value="true" />
@@ -196,7 +200,7 @@
       <workItem from="1727357029326" duration="44000" />
       <workItem from="1727358317602" duration="1867000" />
       <workItem from="1727361217072" duration="594000" />
-      <workItem from="1727434411640" duration="13329000" />
+      <workItem from="1727434411640" duration="14310000" />
     </task>
     <servers />
   </component>
diff --git a/tests/env/optimal_parameters_stmvl_test.toml b/tests/env/optimal_parameters_stmvl_test.toml
new file mode 100644
index 0000000..0f2f0ea
--- /dev/null
+++ b/tests/env/optimal_parameters_stmvl_test.toml
@@ -0,0 +1,4 @@
+[stmvl_test]
+window_size = "2"
+gamma = 0.999999
+alpha = "2"
diff --git a/tests/test_opti_bayesian_cdrec.py b/tests/test_opti_bayesian_cdrec.py
index 60dad9b..c39f708 100644
--- a/tests/test_opti_bayesian_cdrec.py
+++ b/tests/test_opti_bayesian_cdrec.py
@@ -57,5 +57,5 @@ def test_optimization_bayesian_cdrec(self):
 
         Optimization.save_optimization(optimal_params=optimal_params, algorithm=algorithm+"_test")
 
-        self.assertTrue(metrics_optimal["RMSE"] > metrics_default["RMSE"], True, f"Expected {metrics_optimal['RMSE']} > {metrics_default['RMSE']}")
+        self.assertTrue(metrics_optimal["RMSE"] < metrics_default["RMSE"], True, f"Expected {metrics_optimal['RMSE']} > {metrics_default['RMSE']}")
         self.assertTrue(yi > 0, True)
\ No newline at end of file
diff --git a/tests/test_opti_bayesian_iim.py b/tests/test_opti_bayesian_iim.py
index ab778cc..28c6605 100644
--- a/tests/test_opti_bayesian_iim.py
+++ b/tests/test_opti_bayesian_iim.py
@@ -60,5 +60,5 @@ def test_optimization_bayesian_iim(self):
 
         Optimization.save_optimization(optimal_params=optimal_params, algorithm=algorithm+"_test")
 
-        self.assertTrue(metrics_optimal["RMSE"] > metrics_default["RMSE"], True)
+        self.assertTrue(metrics_optimal["RMSE"] < metrics_default["RMSE"], True, f"Expected {metrics_optimal['RMSE']} > {metrics_default['RMSE']}")
         self.assertTrue(yi > 0, True)
\ No newline at end of file
diff --git a/tests/test_opti_bayesian_mrnn.py b/tests/test_opti_bayesian_mrnn.py
index 6125a1c..3bd1d52 100644
--- a/tests/test_opti_bayesian_mrnn.py
+++ b/tests/test_opti_bayesian_mrnn.py
@@ -61,5 +61,5 @@ def test_optimization_bayesian_mrnn(self):
 
         Optimization.save_optimization(optimal_params=optimal_params, algorithm=algorithm+"_test")
 
-        self.assertTrue(metrics_optimal["RMSE"] > metrics_default["RMSE"], True)
+        self.assertTrue(metrics_optimal["RMSE"] < metrics_default["RMSE"], True, f"Expected {metrics_optimal['RMSE']} > {metrics_default['RMSE']}")
         self.assertTrue(yi > 0, True)
\ No newline at end of file
diff --git a/tests/test_opti_bayesian_stmvl.py b/tests/test_opti_bayesian_stmvl.py
index f67ecc5..289a37f 100644
--- a/tests/test_opti_bayesian_stmvl.py
+++ b/tests/test_opti_bayesian_stmvl.py
@@ -61,5 +61,5 @@ def test_optimization_bayesian_stmvl(self):
 
         Optimization.save_optimization(optimal_params=optimal_params, algorithm=algorithm+"_test")
 
-        self.assertTrue(metrics_optimal["RMSE"] > metrics_default["RMSE"], True)
+        self.assertTrue(metrics_optimal["RMSE"] < metrics_default["RMSE"], True, f"Expected {metrics_optimal['RMSE']} > {metrics_default['RMSE']}")
         self.assertTrue(yi > 0, True)
\ No newline at end of file