Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

Appearance settings

rogeroyer/feature_selection_GAAlgorithm

Open more actions menu

Repository files navigation

基于遗传算法做特征选择

运行环境

  • python3.5+
  • numpy
  • pandas
  • matplotlib
  • sklearn
  • LightGBM

使用手册

  • 更换dataSet目录里面的数据集(数据集过大我只上传了一小部分)

    • train_feature.csv : 训练集
    • validate_feature.csv : 验证集
    • 注:两个数据集内的维度要一致。且包含一个标签列,并命名为“target”
  • 更换self.columns为train_feature.csv内的属性名,且第一个元素必须为标签名“target”,其它全为特征名称

  • Genetic_algorithm.py众初始化种群initPopulation那里也要做相应修改

  • 修改self.ga类参数(可选)

  • 修改主函数里的群体个数和迭代次数(可选)


模型及评价指标

  • 评价指标:auc (可修改)
  • 模型:LightGBM (可修改)

结果

程序运行过程会打印出中间过程,最终会绘出迭代次数与最优个体适应图的折线图以及打印出最有个体及其适应度。


Attention

经过两个不同比赛的尝试,评价指标分别是f1_score和auc,线上线下同增减的使用本算法才能收获一个好的结果。

About

基于遗传算法的特征选择

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

Morty Proxy This is a proxified and sanitized view of the page, visit original site.