Skip to content

Commit 5854f2e

Browse files
committed
fix docstring
1 parent 1a42592 commit 5854f2e

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

src/hidimstat/base_variable_importance.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -53,16 +53,18 @@ def selection(
5353
):
5454
"""
5555
Selects features based on variable importance.
56+
5657
Parameters
5758
----------
58-
k_best : int, optional, default=None
59+
k_best : int, default=None
5960
Selects the top k features based on importance scores.
60-
percentile : float, optional, default=None
61+
percentile : float, default=None
6162
Selects features based on a specified percentile of importance scores.
62-
threshold : float, optional, default=None
63+
threshold : float, default=None
6364
Selects features with importance scores above the specified threshold.
64-
threshold_pvalue : float, optional, default=None
65+
threshold_pvalue : float, default=None
6566
Selects features with p-values below the specified threshold.
67+
6668
Returns
6769
-------
6870
selection : array-like of shape (n_features,)
@@ -182,7 +184,7 @@ def selection_fdr(
182184
Raises
183185
------
184186
AssertionError
185-
If test_scores_ is None or if incompatible combinations of parameters are provided
187+
If test_scores\_ is None or if incompatible combinations of parameters are provided
186188
"""
187189
self._check_importance()
188190
assert (

0 commit comments

Comments
 (0)