Skip to content

Commit 661c651

Browse files
OkonSamuelablaom
andauthored
Fix doctests (#19)
* remove dependence of is_wrapper trait on base model. * bug fixes and doc build * fix typo in docstring Co-authored-by: Anthony Blaom, PhD <[email protected]> * fix doctests --------- Co-authored-by: Anthony Blaom, PhD <[email protected]>
1 parent 4f4aa9c commit 661c651

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

docs/src/index.md

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ recursive feature elimination should return the first columns as important featu
2020
```@meta
2121
DocTestSetup = quote
2222
using MLJ, FeatureSelection, StableRNGs
23-
rng = StableRNG(10)
23+
rng = StableRNG(123)
2424
A = rand(rng, 50, 10)
2525
X = MLJ.table(A) # features
2626
y = @views(
@@ -147,11 +147,11 @@ As before we can inspect the important features by inspecting the object returne
147147
```jldoctest
148148
julia> fitted_params(self_tuning_rfe_mach).best_fitted_params.features_left
149149
5-element Vector{Symbol}:
150-
:x1
151-
:x2
152-
:x3
153150
:x4
151+
:x2
152+
:x1
154153
:x5
154+
:x3
155155
156156
julia> feature_importances(self_tuning_rfe_mach)
157157
10-element Vector{Pair{Symbol, Int64}}:
@@ -163,7 +163,7 @@ julia> feature_importances(self_tuning_rfe_mach)
163163
:x3 => 6
164164
:x8 => 5
165165
:x4 => 6
166-
:x10 => 3
166+
:x10 => 3
167167
:x1 => 6
168168
```
169169
and call `predict` on the tuned model machine as shown below

0 commit comments

Comments
 (0)