user670186
user670186

Reputation: 2860

R random forest feature selection based on AUC

For binary option prediction (rise, fall) I am trying random forest in R but the importance measures and OOB are biased in my case

I found this article but it is Python related.

Is there an R package approach for automatic feature selection that

Upvotes: 2

Views: 849

Answers (1)

Manasi Shah
Manasi Shah

Reputation: 437

I just came across this question, I found this package that might help you:

i. It's called AUCRF, it performs feature selection in a random forest model based on optimizing AUC. https://cran.r-project.org/web/packages/AUCRF/AUCRF.pdf

ii. It does allow cross-validation of your AUC based selection AUCRFcv(x, nCV = 5, M = 20)

where nCV is number of folds, M = number of repeats.

iii. Regarding allowing your own evaluation, it does have an option where you can specify the formula using ~ but you will have to explore that more for your specific case, since you have not provided test code.

Hope this helps!

Upvotes: 2

Related Questions