Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
giopaglia authored Jul 5, 2024
1 parent 10f3a2f commit 5d19d3a
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -30,8 +30,6 @@ using MLJ
using MLJDecisionTreeInterface
using DataFrames

using SoleDecisionTreeInterface

X, y = @load_iris
X = DataFrame(X)

Expand All @@ -48,6 +46,8 @@ learned_dt_tree = begin
fitted_params(mach).tree
end

using SoleDecisionTreeInterface

# Convert to Sole model
sole_dt = solemodel(learned_dt_tree)
```
Expand Down

0 comments on commit 5d19d3a

Please sign in to comment.