updated h_selection

rmj3197 · May 30, 2024 · 406a7f8 · 406a7f8
1 parent 06f5d5a
commit 406a7f8
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/QuadratiK/kernel_test/_h_selection.py b/QuadratiK/kernel_test/_h_selection.py
@@ -25,7 +25,7 @@ def _objective_one_sample(
     s_dat,
     skew_data,
     random_state,
-    n_jobs=8,
+    n_jobs=1,
 ):
     """
     Objective function using using the best
@@ -143,7 +143,7 @@ def _objective_two_sample(
     skew_data,
     d,
     random_state,
-    n_jobs=8,
+    n_jobs=1,
 ):
     """
     Objective function using using the best
@@ -287,7 +287,7 @@ def _objective_k_sample(
     s_dat,
     skew_data,
     random_state,
-    n_jobs=8,
+    n_jobs=1,
 ):
     """
     Objective function using using the best
@@ -612,6 +612,7 @@ def select_h(
                     s_dat,
                     skew_data,
                     random_state,
+                    1,
                 )
                 for param in parameters
             )
@@ -677,6 +678,7 @@ def select_h(
                     skew_data,
                     d,
                     random_state,
+                    1,
                 )
                 for param in parameters
             )
@@ -731,6 +733,7 @@ def select_h(
                     s_dat,
                     skew_data,
                     random_state,
+                    1,
                 )
                 for param in parameters
             )

diff --git a/doc/source/user_guide/basic_usage.ipynb b/doc/source/user_guide/basic_usage.ipynb
@@ -390,7 +390,7 @@
     "\n",
     "# Perform the algorithm for selecting h\n",
     "h_selected, all_powers = select_h(\n",
-    "    x=X1, y=X2, alternative=\"skewness\", method=\"subsampling\", b=0.2\n",
+    "    x=X1, y=X2, alternative=\"skewness\", method=\"subsampling\", b=0.5, n_jobs=-1\n",
     ")\n",
     "print(\"Selected h is: {}\".format(h_selected))\n",
     "\n",
@@ -429,7 +429,7 @@
     "\n",
     "# Perform the algorithm for selecting h\n",
     "h_selected, all_powers = select_h(\n",
-    "    x=X, y=y, alternative=\"skewness\", n_jobs=-1, b=0.2, method=\"subsampling\"\n",
+    "    x=X, y=y, alternative=\"skewness\", n_jobs=-1, b=0.5, method=\"subsampling\"\n",
     ")\n",
     "print(\"Selected h is: {}\".format(h_selected))\n",
     "\n",