R Language
Apprentissage automatique

Créer un modèle de forêt aléatoire

Un exemple d'algorithme d'apprentissage automatique est l'alogrithme de forêt aléatoire (Breiman, L. (2001). Random Forests. Machine Learning 45 (5) , p. 5-32). Cet algorithme est implémenté dans R conformément à l'implémentation originale de Breiman dans Fortran dans le package randomForest .

Les objets classificateurs de forêt aléatoire peuvent être créés dans R en préparant la variable de classe en tant que factor , ce qui apparaît déjà dans le jeu de données du iris . Par conséquent, nous pouvons facilement créer une forêt aléatoire en:

library(randomForest)

rf <- randomForest(x = iris[, 1:4], 
                   y = iris$Species, 
                   ntree = 500, 
                   do.trace = 100)

rf

# Call:
#   randomForest(x = iris[, 1:4], y = iris$Species, ntree = 500,      do.trace = 100) 
# Type of random forest: classification
# Number of trees: 500
# No. of variables tried at each split: 2
# 
# OOB estimate of  error rate: 4%
# Confusion matrix:
#   setosa versicolor virginica class.error
# setosa         50          0         0        0.00
# versicolor      0         47         3        0.06
# virginica       0          3        47        0.06