bâtiment de l'arbre de classification ayant variables catégorielles à l'aide de rpart

J'ai un jeu de données avec des fonctionnalités 14 et peu d'entre eux sont comme ci-dessous, où le sexe et l'état matrimonial sont des variables catégorielles.

height,sex,maritalStatus,age,edu,homeType

SEX
         1. Male
         2. Female

MARITAL STATUS
         1. Married
         2. Living together, not married
         3. Divorced or separated
         4. Widowed
         5. Single, never married

Maintenant, je suis en utilisant rpart bibliothèque de R à construire un arbre de classification à l'aide de la suite de

rfit = rpart(homeType ~., data = trainingData, method = "class", cp = 0.0001)

Cela me donne un arbre de décision qui ne considère pas le sexe et l'état matrimonial comme facteurs.

Je pense de l'utilisation que.facteur :

sex = as.factor(trainingData$sex)
ms = as.factor(trainingData$maritalStatus)

Mais je ne suis pas sûr de savoir comment dois-je passer cette information à rpart. Puisque les données de l'argument dans rpart() prend en "trainingData" bloc de données. Il y en aura toujours les valeurs qui sont dans ce bloc de données.
Je suis un peu nouveau pour R et vous serions reconnaissants de l'aide de quelqu'un sur ce.

OriginalL'auteur user4251309 | 2014-11-14

  1. 7

    Vous pouvez faire les modifications à la trainingData bloc de données directement, puis exécutez rpart().

    trainingData$sex = as.factor(trainingData$sex)
trainingData$maritalStatus = as.factor(trainingData$maritalStatus)
rfit = rpart(homeType ~., data = trainingData, method = "class", cp = 0.0001)
    Je suis en train d'appliquer cette réponse pour un exemple similaire où ma variable catégorique c'est les jours de la semaine. Juste pour être sûr que j'ai quitté que cette variable dans l'ensemble d'apprentissage, mais lorsque j'essaie de former le classificateur, le modèle semble avoir un seul nœud racine, ce qui signifie que je ne prend pas la variable en compte. Avez-vous une idée de ce que pourrait être le problème?
    Voir stackoverflow.com/a/20994978/2140956.

    OriginalL'auteur Jean V. Adams

  2. -3

    Dans la pratique, vous pouvez transformer n'importe quel catégorique valeur en une valeur ordinale, par exemple "État Matrimonial" dans les conditions 1, 2, 3... Mais, en général, vous ne devriez pas faire la transformation, sauf si vous avez une définition conceptuelle de toute valeur continue. Par exemple, si vous ne pouvez pas définir ce qu'est une 1.2 Martital État, vous ne devriez pas faire la transformation. Au lieu de cela, parfois, vous pouvez utiliser une valeur représentative, selon l'objectif de votre recherche. Par exemple, si vous tentez de lier vos données pour prédire le type de maison, le "degré minimum de confort" de chaque situation de famille est une valeur ordinale qui est en mesure d'être interprétée si (disons) est de 1,2.

    OriginalL'auteur Jose Carlos Machicao Valencia