1. Apprendre
    2. /
  2. Cours
    3. /
  3. Introduction à Spark avec sparklyr en R
Connected

Exercice

Random Forest : prédiction

Vous devez maintenant générer des prédictions avec votre modèle de random forest. La syntaxe est la même que pour le modèle de gradient boosted trees.

Instructions

100 XP

Une connexion Spark a été créée pour vous sous le nom spark_conn. Les tibbles associés aux ensembles d'entraînement et de test stockés dans Spark ont été prédéfinis comme track_data_to_model_tbl et track_data_to_predict_tbl, respectivement. Le modèle de random forest a été prédéfini sous random_forest_model.

  • Définissez une variable predicted qui contient les prédictions du modèle pour nos données de test.
    • Appelez ml_predict() avec le modèle et les données de test comme arguments. Cette fonction génère des prédictions pour l'ensemble de test et les ajoute comme nouvelle colonne nommée prediction.
  • Définissez la variable responses pour préparer les données en vue de comparer les réponses prédites aux réponses réelles :
    • Sélectionnez la colonne de réponse year.
    • Collectez les résultats.
    • Utilisez mutate() pour ajouter les prédictions contenues dans predicted.