Zacznij terazZacznij za darmo

Tworzenie wykresu pudełkowego

caret oferuje wiele metod porównywania modeli. Wszystkie opierają się na funkcji resamples(). Szczególnie przydatny jest wykres pudełkowy (box-and-whisker), który pozwala porównać rozkład dokładności predykcji (w tym przypadku AUC) dla dwóch modeli.

Zasadniczo zależy ci na modelu z wyższą medianą AUC oraz mniejszym rozstępem między wartością minimalną a maksymalną AUC.

Wykres ten można utworzyć za pomocą funkcji bwplot(), która generuje wykres pudełkowy wyników modelu na danych spoza zbioru treningowego. Na wykresie pudełkowym mediana każdego rozkładu jest zaznaczona linią, a rozstęp międzykwartylowy – prostokątem wokół tej linii. Możesz przekazać argument metric = "ROC" do funkcji bwplot(), aby wyświetlić wyniki ROC modelu na danych spoza zbioru treningowego i wybrać model z najwyższą medianą ROC.

Jeśli nie podasz metryki do wykreślenia, bwplot() automatycznie wyświetli 3 metryki.

To ćwiczenie jest częścią kursu

Uczenie maszynowe z caret w R

Zobacz kurs

Instrukcje do ćwiczenia

Przekaż obiekt resamples do funkcji bwplot(), aby utworzyć wykres pudełkowy. Przyjrzyj się wynikowi i zwróć uwagę, który model ma wyższą medianę statystyki ROC. Pamiętaj, aby wskazać, którą metrykę chcesz wykreślić.

Interaktywne ćwiczenie praktyczne

Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.

# Create bwplot
Edytuj i uruchom kod