daviddoji/JuliaForDataAnalysis
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update codes to include confusion matrix

Browse Source
This commit is contained in:
Bogumił Kamiński
2022-02-26 16:02:00 +01:00
parent f955d30252
commit 65977e8d22
1 changed files with 13 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
ch13.jl
View File

@@ -244,15 +244,21 @@ histogram!(test_groups[(true,)].predict;
bins=10, normalize=:probability, bins=10, normalize=:probability,
fillalpha=0.5, label="true") fillalpha=0.5, label="true")
# Code for computing of confusion matrix
@chain test begin
@rselect(:observed=:arrest, :predicted=:predict > 0.15)
proptable(:predicted, :observed; margins=2)
end
# Code for listing 13.12 # Code for listing 13.12
test_roc = roc(test, score=:predict, target=:arrest) test_roc = roc(test, score=:predict, target=:arrest)
plot(test_roc.pfa, 1 .- test_roc.pmiss; plot(test_roc.pfa, test_roc.pmiss;
legend=:bottomright,
color="black", lw=3, color="black", lw=3,
label="test (AUC=$(round(100*(1-auc(test_roc)), digits=2))%)", label="test (AUC=$(round(100*auc(test_roc), digits=2))%)",
xlabel="FPR", ylabel="TPR") xlabel="pfa", ylabel="pmiss")
train_roc = roc(train, score=:predict, target=:arrest) train_roc = roc(train, score=:predict, target=:arrest)
plot!(train_roc.pfa, 1 .- train_roc.pmiss; plot!(train_roc.pfa, train_roc.pmiss;
color="gold", lw=3, color="gold", lw=3,
label="train (AUC=$(round(100*(1-auc(train_roc)), digits=2))%)") label="train (AUC=$(round(100*auc(train_roc), digits=2))%)")