summaryrefslogtreecommitdiff
path: root/images/featureselectionlizeo.R
diff options
context:
space:
mode:
Diffstat (limited to 'images/featureselectionlizeo.R')
-rw-r--r--images/featureselectionlizeo.R20
1 files changed, 20 insertions, 0 deletions
diff --git a/images/featureselectionlizeo.R b/images/featureselectionlizeo.R
new file mode 100644
index 0000000..7df6b8c
--- /dev/null
+++ b/images/featureselectionlizeo.R
@@ -0,0 +1,20 @@
+#!/usr/bin/env Rscript
+load (sprintf ("%s/data/feature_selection_lizeo.Rdata", Sys.getenv ("ABS_TOP_SRCDIR")))
+Sys.setlocale ("LC_ALL", "fr_FR.UTF-8")
+library ("magrittr")
+plot <- (feature_selection_lizeo
+ %>% dplyr::filter (algorithm != "rsms_full")
+ %>% dplyr::mutate (algorithm = ifelse (algorithm == "rsms", "\\textbf{RSMS}", algorithm))
+ %>% dplyr::mutate (algorithm = ifelse (algorithm == "mifs", "MIFS", algorithm))
+ %>% dplyr::mutate (algorithm = ifelse (algorithm == "rfs", "RFS", algorithm))
+ %>% dplyr::mutate (`% de variables sélectionnées` = n_features / max (n_features))
+ %>% dplyr::select (`Algorithme` = algorithm, `% de variables sélectionnées`, `aRMSE` = armse)
+ %>% dplyr::group_by (Algorithme, `% de variables sélectionnées`)
+ %>% dplyr::summarize (aRMSE = min (aRMSE))
+ %>% ggplot2::ggplot (ggplot2::aes (x = `% de variables sélectionnées`, y = aRMSE, color = Algorithme, linetype = Algorithme))
+ + ggplot2::geom_line ()
+ + ggplot2::scale_linetype_manual (values = c ("\\textbf{RSMS}" = "solid", "MIFS" = "dashed", "RFS" = "twodash"))
+ + ggplot2::scale_color_manual (values = c ("\\textbf{RSMS}" = "#E69F00", "MIFS" = "#56B4E9", "RFS" = "#009E73"))
+ + ggplot2::scale_x_continuous (trans = 'log10', labels = scales::percent))
+filename <- Sys.getenv ("OUTPUT")
+ggplot2::ggsave (filename, plot, device = "svg", width = 6, height = 3)