3. adding of eeg dataset with bayesian tests

.github/workflows/pytest_10.yml

@@ -23,7 +23,7 @@ jobs:
         sudo apt-get update
         sudo apt-get install libmlpack-dev
         sudo apt-get install libopenblas-dev
-        sudo apt-get install python3-dev
+        sudo apt-get install python3-dev build-essential
         pip install -r requirements.txt
         pip install mypy
         pip install pytest

.github/workflows/pytest_contamination.yml

@@ -23,7 +23,7 @@ jobs:
         sudo apt-get update
         sudo apt-get install libmlpack-dev
         sudo apt-get install libopenblas-dev
-        sudo apt-get install python3-dev
+        sudo apt-get install python3-dev build-essential
         pip install -r requirements.txt
         pip install mypy
         pip install pytest

.github/workflows/pytest_imp_cdrec.yml

@@ -24,7 +24,7 @@ jobs:
         sudo apt-get update
         sudo apt-get install libmlpack-dev
         sudo apt-get install libopenblas-dev
-        sudo apt-get install python3-dev
+        sudo apt-get install python3-dev build-essential
         pip install -r requirements.txt
         pip install mypy
         pip install pytest

.github/workflows/pytest_imp_iim.yml

@@ -24,7 +24,7 @@ jobs:
         sudo apt-get update
         sudo apt-get install libmlpack-dev
         sudo apt-get install libopenblas-dev
-        sudo apt-get install python3-dev
+        sudo apt-get install python3-dev build-essential
         pip install -r requirements.txt
         pip install mypy
         pip install pytest

.github/workflows/pytest_imp_mrnn.yml

@@ -24,7 +24,7 @@ jobs:
         sudo apt-get update
         sudo apt-get install libmlpack-dev
         sudo apt-get install libopenblas-dev
-        sudo apt-get install python3-dev
+        sudo apt-get install python3-dev build-essential
         pip install -r requirements.txt
         pip install mypy
         pip install pytest

.github/workflows/pytest_imp_stmvl.yml

@@ -24,7 +24,7 @@ jobs:
         sudo apt-get update
         sudo apt-get install libmlpack-dev
         sudo apt-get install libopenblas-dev
-        sudo apt-get install python3-dev
+        sudo apt-get install python3-dev build-essential
         pip install -r requirements.txt
         pip install mypy
         pip install pytest

.github/workflows/pytest_loading.yml

@@ -23,7 +23,7 @@ jobs:
         sudo apt-get update
         sudo apt-get install libmlpack-dev
         sudo apt-get install libopenblas-dev
-        sudo apt-get install python3-dev
+        sudo apt-get install python3-dev build-essential
         pip install -r requirements.txt
         pip install mypy
         pip install pytest

.github/workflows/pytest_opt_cdrec.yml

@@ -24,7 +24,7 @@ jobs:
         sudo apt-get update
         sudo apt-get install libmlpack-dev
         sudo apt-get install libopenblas-dev
-        sudo apt-get install python3-dev
+        sudo apt-get install python3-dev build-essential
         pip install -r requirements.txt
         pip install mypy
         pip install pytest

.github/workflows/pytest_opt_iim.yml

@@ -24,7 +24,7 @@ jobs:
         sudo apt-get update
         sudo apt-get install libmlpack-dev
         sudo apt-get install libopenblas-dev
-        sudo apt-get install python3-dev
+        sudo apt-get install python3-dev build-essential
         pip install -r requirements.txt
         pip install mypy
         pip install pytest

.github/workflows/pytest_opt_mrnn.yml

@@ -24,7 +24,7 @@ jobs:
         sudo apt-get update
         sudo apt-get install libmlpack-dev
         sudo apt-get install libopenblas-dev
-        sudo apt-get install python3-dev
+        sudo apt-get install python3-dev build-essential
         pip install -r requirements.txt
         pip install mypy
         pip install pytest

.github/workflows/pytest_opt_stmvl.yml

@@ -24,7 +24,7 @@ jobs:
         sudo apt-get update
         sudo apt-get install libmlpack-dev
         sudo apt-get install libopenblas-dev
-        sudo apt-get install python3-dev
+        sudo apt-get install python3-dev build-essential
         pip install -r requirements.txt
         pip install mypy
         pip install pytest

.github/workflows/pytest_shap_explainer.yml

@@ -24,7 +24,7 @@ jobs:
         sudo apt-get update
         sudo apt-get install libmlpack-dev
         sudo apt-get install libopenblas-dev
-        sudo apt-get install python3-dev
+        sudo apt-get install python3-dev build-essential
         pip install -r requirements.txt
         pip install mypy
         pip install pytest

imputegap/imputation/imputation.py

@@ -79,7 +79,7 @@ def evaluate_params(ground_truth, contamination, configuration, algorithm="cdrec
             imputation, error_measures = Imputation.MR.cdrec(ground_truth, contamination, (rank, eps, iters))
         elif algorithm == 'iim':
             learning_neighbours = configuration
-            alg_code = "iim " + re.sub(r'[\W_]', '', str(learning_neighbours))
+            alg_code = "iim " + learning_neighbours
             imputation, error_measures = Imputation.Regression.iim_imputation(ground_truth, contamination, (learning_neighbours, alg_code))
         elif algorithm == 'mrnn':
             hidden_dim, learning_rate, iterations, keep_prob, seq_len = configuration

requirements.txt

@@ -1,4 +1,4 @@
-numpy==1.21.5
+numpy==1.13.3
 pandas==2.0.3
 matplotlib==3.7.5
 toml==0.10.2
@@ -8,6 +8,7 @@ setuptools==75.1.0
 tensorflow==2.13.1
 shap==0.44.1
 pycatch22==0.4.5
+scikit-optimize==0.10.2
 types-toml
 types-setuptools
 wheel

tests/test_opti_bayesian_iim.py

@@ -34,7 +34,7 @@ def get_file_path(set_name="test"):
 
 class TestOptiIMM(unittest.TestCase):
 
-    def test_optimization_bayesian_stmvl(self):
+    def test_optimization_bayesian_iim(self):
         """
         the goal is to test if only the simple optimization with stmvl has the expected outcome
         """

tests/test_opti_bayesian_mrnn.py

@@ -32,9 +32,9 @@ def get_file_path(set_name="test"):
     return resolve_path(f'../imputegap/dataset/{set_name}.txt', f'./imputegap/dataset/{set_name}.txt')
 
 
-class TestOptiSTMVL(unittest.TestCase):
+class TestOptiMRNN(unittest.TestCase):
 
-    def test_optimization_bayesian_stmvl(self):
+    def test_optimization_bayesian_mrnn(self):
         """
         the goal is to test if only the simple optimization with mrnn has the expected outcome
         """