From a5a489a7494779fbbfc46f7740bec4c81042bd24 Mon Sep 17 00:00:00 2001 From: Loic Guegan Date: Fri, 25 Nov 2022 14:43:10 +0100 Subject: Update --- analysis/learning.R | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) (limited to 'analysis/learning.R') diff --git a/analysis/learning.R b/analysis/learning.R index ff31d33..684dc7b 100644 --- a/analysis/learning.R +++ b/analysis/learning.R @@ -101,14 +101,17 @@ build_models=function(ignore_hint=TRUE){ ## Prefs f1_knn=F1_Score2(test_set$simkey,knn_predictions) f1_tree=F1_Score2(test_set$simkey,tree_predictions) - list(cbind(tibble(model=c("knn","tree")),rbind(f1_knn,f1_tree))) + accuracy_knn=sum(test_set$simkey==knn_predictions)/length(test_set$simkey) + accuracy_tree=sum(test_set$simkey==tree_predictions)/length(test_set$simkey) + list(cbind(tibble(model=c("knn","tree")),rbind(f1_knn,f1_tree),tibble(accuracy=c(accuracy_knn,accuracy_tree)))) }) perfs=do.call("rbind",perfs)%>%mutate_if(is.numeric, ~round(.,digits=2)) perfs=perfs%>%group_by(model)%>%summarize( f1_baseline=mean(f1_baseline), f1_hint=mean(f1_hint), f1_extended=mean(f1_extended), - f1_hintandextended=mean(f1_hintandextended)) + f1_hintandextended=mean(f1_hintandextended), + accuracy=mean(accuracy)) write.csv(perfs,"figures/f1_scores_offline.csv",quote=FALSE,row.names=FALSE) ## Return models -- cgit v1.2.3