Skip to content

Commit 583324b

Browse files
Merge pull request #16 from MatthewSZhang/doc-readme-example
DOC add multioutput example in readme
2 parents f09b191 + 50ad1b8 commit 583324b

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

README.rst

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ Getting Started
6161
... [-2.79, -0.02, -0.85 ],
6262
... [-1.34, -0.48, -2.55 ],
6363
... [ 1.92, 1.48, 0.65 ]]
64-
>>> y = [0, 1, 0, 1]
64+
>>> y = [[0, 0], [1, 1], [0, 0], [1, 0]] # Multioutput feature selection
6565
>>> selector = FastCan(n_features_to_select=2, verbose=0).fit(X, y)
6666
>>> selector.get_support()
6767
array([ True, True, False])
@@ -70,14 +70,14 @@ array([0, 1])
7070
>>> selector.indices_ # Indices in selection order
7171
array([1, 0], dtype=int32)
7272
>>> selector.scores_ # Scores for selected features in selection order
73-
array([0.64276838, 0.09498243])
73+
array([0.91162413, 0.71089547])
7474
>>> # Here Feature 2 must be included
7575
>>> selector = FastCan(n_features_to_select=2, indices_include=[2], verbose=0).fit(X, y)
7676
>>> # We can find the feature which is useful when working with Feature 2
7777
>>> selector.indices_
78-
array([2, 1], dtype=int32)
78+
array([2, 0], dtype=int32)
7979
>>> selector.scores_
80-
array([0.16632562, 0.50544788])
80+
array([0.34617598, 0.95815008])
8181

8282

8383
Citation

0 commit comments

Comments
 (0)