sparklyr: sélectionne toutes les colonnes sauf réponse lors de la formation du modèle

Question

J'ai un Spark DataFrame, train_tbl, qui contient 48 colonnes différentes. Je veux former un modèle de forêt aléatoire avec le package sparklyr qui utilise l'une des 48 colonnes en tant que variable response et toutes les autres colonnes en tant que features. Existe-t-il un moyen de spécifier que je veux utiliser tous les champs sauf loan_status comme fonctionnalités, sans avoir à taper tous les 47 noms de champs?

ml_random_forest(x = train_tbl, 
       response = "loan_status", 
       features = call all fields EXCEPT "loan_status", 
       num.trees = 10L, 
       type = "classification") 

Answer 1

Cela devrait fonctionner.

ml_random_forest(x = train_tbl, 
       response = "loan_status", 
       features = names(train_tbl)[which(names(train_tbl)!="loan_status")], 
       num.trees = 10L, 
       type = "classification")