Update

1 files changed, 5 insertions, 2 deletions

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