[R] randomForest: How to append ID column along with predictions
Liaw, Andy
andy_liaw at merck.com
Tue Dec 7 18:00:04 CET 2010
The order in the output correspond to the order of the input. I will
patch the code so that it grabs the row names of the input (if exist).
If you specify type="prob", it already labels the rows by the input row
names.
> -----Original Message-----
> From: r-help-bounces at r-project.org
> [mailto:r-help-bounces at r-project.org] On Behalf Of Dennis Duro
> Sent: Tuesday, December 07, 2010 11:46 AM
> To: r-help at r-project.org
> Subject: [R] randomForest: How to append ID column along with
> predictions
>
> Hi all,
>
> When running a prediction using RF on another data, I get two columns
> returned: row number(?) and predicted class. Is there a way of
> appending the unique row value from an ID column in the dataframe to
> the predictions instead of the row number? I'm assuming that the
> returned results follow the data frame in that the first result
> returned equals the first entry in the dataframe.
>
> i.e., instead of a prediction output like this:
>
> 1, ants
> 2, ants
> 3, bees
> 4, ants
>
> I'd like the first column to pull IDs from the dataframe associated
> with each row (row number in parenthesis for illustration):
>
> (1) 1130, ants
> (2) 1130, ants
> (3) 2139, bees
> (4) 1130, ants
>
> This is likely a simple procedure, but I haven't been able to get
> anything to work. Any help would be appreciated!
>
> Cheers,
>
> Dennis
>
> ______________________________________________
> R-help at r-project.org mailing list
> https://stat.ethz.ch/mailman/listinfo/r-help
> PLEASE do read the posting guide
> http://www.R-project.org/posting-guide.html
> and provide commented, minimal, self-contained, reproducible code.
>
Notice: This e-mail message, together with any attachme...{{dropped:11}}
More information about the R-help
mailing list