Add example with cutoff again to demonstrate idiom detection potential
Change-Id: I85be4ecffabb9a0cb98091b678c71991230591d3
diff --git a/R/idiomclassification_mk_pf.R b/R/idiomclassification_mk_pf.R
index c1f4dbc..7a591b7 100644
--- a/R/idiomclassification_mk_pf.R
+++ b/R/idiomclassification_mk_pf.R
@@ -82,6 +82,13 @@
res <- confusionMatrix(prediction_for_table,test$CO_IDIOM, positive = "1")
print(res)
+cat("With SMOTE and detection task oriented cutoff for prediction\n")
+
+prediction_for_table <- predict(rf_classifier,test %>% select(-CO_IDIOM), cutoff = c(0.8, 0.2))
+res <- confusionMatrix(prediction_for_table,test$CO_IDIOM, positive = "1")
+print(res)
+
+
# Using estimates by random forest on entire dataset
library(randomForest)