10. adding of eeg dataset with bayesian tests

eXascaleInfolab · Sep 27, 2024 · 54e109c · 54e109c
1 parent 043bc5a
commit 54e109c
Show file tree

Hide file tree

Showing 6 changed files with 14 additions and 6 deletions.
diff --git a/.idea/workspace.xml b/.idea/workspace.xml
diff --git a/tests/env/optimal_parameters_stmvl_test.toml 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
@@ -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)
diff --git 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)
diff --git 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)
diff --git 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)