Skip to content
Basic data mining model, including feature importance display
Python
Branch: master
Clone or download
Latest commit 70ae752 Aug 22, 2019
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
code modify some bug Aug 18, 2019
data update README Aug 3, 2019
result first commit Aug 3, 2019
README.md update README Aug 22, 2019
requirements.txt add pytorch Aug 13, 2019

README.md

数据挖掘类比赛常用模型

1. 概述

该工程主要是几个比较常用的模型,包含二分类模型、多分类模型以及回归模型。他们分别是基于lightgbm实现、xgboost实现和keras实现:

2.环境设置

可以直接通过pip install -r requirements.txt安装指定的函数包,具体的函数包如下:

pandas
numpy
matplotlib
sklearn
tensorflow==1.12.0
keras==2.2.4
pytorch
seaborn
lightgbm==2.2.1
xgboost==0.90

3. 代码讲解

4. 最后

目前只整理了几个常用的模型,下一步会将用到的不错的代码同步进来,如果有问题或者发现有什么错误的地方需要交流可以通过微信公众号或者代码中的邮箱联系作者,也可以通过Issues提出问题。如果该项目对你有帮助,麻烦点击一下star,鼓励一下作者。如果有最新内容更新会第一时间在微信公众号内发布,希望能和你一起在AI方面前进,成长。

作者的知乎:QLMX, 知乎专栏:ML与DL成长之路

微信公号:AI成长社

You can’t perform that action at this time.