[R] To write a function for getting the node numbers in "rpart" analysis..

Yashwanth M.R yashwanth at predictiveanalytics.in
Thu Feb 9 12:25:01 CET 2012


## To read .csv file(data) and performing "rpart" ##
# library(Hmisc)
library(rpart)

read.csv("D:/Training/RRE/Revolution(My Projects)/Project1/RProject(pmml
from SPSS)/telco_churn.csv", header = TRUE, sep = ",", quote="\"", dec=".")
# help(read.csv)
Telco<- read.table(file = "D:/Training/RRE/Revolution(My
Projects)/Project1/RProject(pmml from SPSS)/telco_churn.csv", header = TRUE,
sep = ",")
names(Telco)

##################################
##################################

## To make factor variable ##
Telco$region<-factor(Telco$region,levels=c(1,2,3,4,5),labels=c("Zone1","Zone2","Zone3","Zone4","Zone5"))

Telco$ed<-factor(Telco$ed,levels=c(1,2,3,4,5),labels=c("DidNotHighSchool","HighSchoolDegree","SomeCollege",
							"CollegeDegree","PostUndergraduateDegree"))

Telco$marital<-factor(Telco$marital,,levels=c(0,1),labels=c("Married","Un-married"))

Telco$churn<-factor(Telco$churn,levels=c(0,1),labels=c("NO","YES"))

##################################
##################################

## To perform "rpart" ##

Telco.rpart <- rpart(churn ~
region+ed+marital+tenure+age+address+income,data=Telco,method="class")
plot(Telco.rpart,compress=FALSE,uniform=TRUE)
text(Telco.rpart,use.n = TRUE, cex = .75) 

##################################
##################################

## To predict "rpart" ##

TelcoPredicted <- predict(Telco.rpart,type="class") # Prediction/Scoring


##################################
##################################





In the above computation, I would prefer to get the list of "node numbers"
parallely  for the each of the predictions of "TelcoPredicted". Please help
me writing a function for this.

--
View this message in context: http://r.789695.n4.nabble.com/To-write-a-function-for-getting-the-node-numbers-in-rpart-analysis-tp4372508p4372508.html
Sent from the R help mailing list archive at Nabble.com.



More information about the R-help mailing list