Skip to content

pablo14/genetic-algorithm-feature-selection

master
Switch branches/tags

Name already in use

A tag already exists with the provided branch name. Many Git commands accept both tag and branch names, so creating this branch may cause unexpected behavior. Are you sure you want to create this branch?
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 

Feature Selection using Genetic Algorithms in R

This script select the 'best' subset of variables based on genetic algorithms in R.

It uses a custom fitness function for binary-class classification. Please modify it to use in other scenarios.

This sctipt is realated to the blog post: Feature Selection using Genetic Algorithms in R.

How to run the example?

The code is ready to calculate the best subset for a cancer dataset (data_breast_cancer2.csv), included in the repo. This is in var_sel_gen_alg.R.

The initial data preparation removes the NA, and it converts the target variable (data_y) into a factor in order to create the predictive model. Both are conditions for the Random Forest that is built behind, using caret package.

You can find how the predictive model is created in the function get_roc_metric, inside the lib_ga.R file. And it is, as the name suggests, prepared for a binary classification problem.

My suggestion before applying with your data is, use the function get_roc_metric directly with your current dataset, and see if it ends ok. If so, proceed to apply the GA procedure.

My idea was to encapsulate the fitness function so you can adjust it better for your purposes. For example, if you want to return Kappa statistic (multi-class target), or RMSE (regression-target). You have to replace get_roc_metric function.

--

Feel free to share your insights, have fun!

Twitter

About

Script to select the best subset of variables based on genetic algorithm in R

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages