4.9. Random forest visualization

library(randomForest)

## randomForest 4.7-1.1

## Type rfNews() to see new features/changes/bug fixes.

library(randomForestExplainer)

## Registered S3 method overwritten by 'GGally':
##   method from   
##   +.gg   ggplot2

library(AER)

## Загрузка требуемого пакета: car

## Загрузка требуемого пакета: carData

## Загрузка требуемого пакета: lmtest

## Загрузка требуемого пакета: zoo

## 
## Присоединяю пакет: 'zoo'

## Следующие объекты скрыты от 'package:base':
## 
##     as.Date, as.Date.numeric

## Загрузка требуемого пакета: sandwich

## Загрузка требуемого пакета: survival

library(ggplot2)

## 
## Присоединяю пакет: 'ggplot2'

## Следующий объект скрыт от 'package:randomForest':
## 
##     margin

data("TeachingRatings")
TeachingRatings$prof <- as.numeric(TeachingRatings$prof)
forest <- randomForest(eval ~., data = TeachingRatings, localImp = TRUE)
min_depth_frame <- min_depth_distribution(forest)
plot_min_depth_distribution(min_depth_frame, k = 8) + 
  xlab("Переменные") +
  ylab("Количество деревьев")+ theme_classic() + 
  scale_fill_manual(values = c("#003f5c", "#2f4b7c", "#665191", "#a05195", "#d45087", "#f95d6a", "#ff7c43", "#ffa600", 
                               "#e45149", "#b6d2dd", "#259086", "#0a463c", "darkblue", "darkred", "#665444", "#669444", "#678399", "#682901"))