Skip to content

Commit

Permalink
Print models in binary classification example better
Browse files Browse the repository at this point in the history
  • Loading branch information
rikhuijzer committed Sep 26, 2023
1 parent 21d6ae6 commit cab8250
Show file tree
Hide file tree
Showing 2 changed files with 15 additions and 7 deletions.
1 change: 1 addition & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@ deps/src/
# Build artifacts for creating documentation generated by the Documenter package
docs/build/
docs/site/
docs/src/*.md

# File generated by Pkg, the package manager, based on a corresponding Project.toml
# It records a fixed state of all packages used by the project. As such, it should not be
Expand Down
21 changes: 14 additions & 7 deletions docs/src/binary-classification.jl
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
### A Pluto.jl notebook ###
# v0.19.27
# v0.19.28

using Markdown
using InteractiveUtils
Expand Down Expand Up @@ -631,30 +631,30 @@ end
# ╠═╡ show_logs = false
e1 = let
model = DecisionTreeClassifier
hyperparameters = (; max_depth=2, rng=_rng())
hyperparameters = (; rng=_rng(), max_depth=2)
_evaluate(model, hyperparameters, X, y)
end;

# ╔═╡ 6ea43d21-1cc0-4bca-8683-dce67f592949
# ╠═╡ show_logs = false
e2 = let
model = StableRulesClassifier
hyperparameters = (; q=4, max_depth=2, max_rules=8, rng=_rng())
hyperparameters = (; rng=_rng(), q=4, max_depth=2, max_rules=8)
_evaluate(model, hyperparameters, X, y)
end;

# ╔═╡ 88a708a7-87e8-4f97-b199-70d25ba91894
# ╠═╡ show_logs = false
e3 = let
model = StableRulesClassifier
hyperparameters = (; q=4, max_depth=2, max_rules=25, rng=_rng())
hyperparameters = (; rng=_rng(), q=4, max_depth=2, max_rules=25)
_evaluate(model, hyperparameters, X, y)
end;

# ╔═╡ 86ed4d56-23e6-4b4d-9b55-7067124da27f
e4 = let
model = StableRulesClassifier
hyperparameters = (; q=4, max_depth=1, max_rules=25, rng=_rng())
hyperparameters = (; rng=_rng(), q=4, max_depth=1, max_rules=25)
_evaluate(model, hyperparameters, X, y)
end;

Expand All @@ -670,7 +670,7 @@ fitresults = getproperty.(e4.e.fitted_params_per_fold, :fitresult);
# ╠═╡ show_logs = false
e5 = let
model = StableForestClassifier
hyperparameters = (; q=4, max_depth=2, rng=_rng())
hyperparameters = (; rng=_rng(), q=4, max_depth=2)
_evaluate(model, hyperparameters, X, y)
end;

Expand All @@ -684,7 +684,7 @@ end;
# ╔═╡ 8c093e66-f476-4501-b098-705e274be8ee
e7 = let
model = EvoTreeClassifier
hyperparameters = (; max_depth=2, rng=_rng())
hyperparameters = (; rng=_rng(), max_depth=2)
_evaluate(model, hyperparameters, X, y)
end;

Expand All @@ -698,6 +698,13 @@ results = let
text = string(score)
length(text) < 4 ? text * '0' : text
end
df[:, :Model] = map(zip(df.Model, df.Hyperparameters)) do (model, param)
text = string(model, param)
text = replace(text, " = " => "=")
text = replace(text, ",)" => ")")
text = replace(text, "(;)" => "()")
end
select!(df, Not(:Hyperparameters))
for col in names(df)
df[!, col] = Base.Text.(df[:, col])
end
Expand Down

0 comments on commit cab8250

Please sign in to comment.