diff options
| -rw-r--r-- | analysis/offline.R | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/analysis/offline.R b/analysis/offline.R index da444d8..c63ddbe 100644 --- a/analysis/offline.R +++ b/analysis/offline.R @@ -59,7 +59,7 @@ build_models=function(ignore_hint=TRUE){ knn_fn=function(inputs){ as.vector(knn(train=train_set%>%select(-simkey),test=inputs%>%select(-simkey),cl=train_set$simkey,k=10)) } - + ## Decision tree tree=rpart( simkey ~ wireless + wakeupfor + energy + coverage, @@ -113,7 +113,7 @@ build_models=function(ignore_hint=TRUE){ f1_hintandextended=mean(f1_hintandextended), accuracy=mean(accuracy)) write.csv(perfs,paste0("figures/f1_scores_offline_ignoreHINT",ignore_hint,".csv"),quote=FALSE,row.names=FALSE) - + ## Return models models } @@ -160,7 +160,7 @@ generate_inputs=function(ignore_hint=FALSE) { elbow_data=do.call("rbind",elbow_data) ggplot(data=elbow_data,aes(k,accuracy))+geom_point()+geom_line()+ggtitle(paste("K-elbow for with NoHint to",as.character(ignore_hint)))+ylim(c(0,1)) ggsave(paste0("figures/knn_elbow_NoHintIs",as.character(ignore_hint),".pdf")) - + ## Prints print(paste0("Accuracy: KNN=",knn_accuracy,"% CART=",tree_accuracy,"%")) pdf(paste0("figures/tree_",as.character(ignore_hint),".pdf")) @@ -210,5 +210,5 @@ generate_inputs=function(ignore_hint=FALSE) { } ## Generate inputs -generate_inputs(FALSE) -generate_inputs(TRUE) +generate_inputs(ignore_hint=FALSE) +generate_inputs(ignore_hint=TRUE) |