数据挖掘实战(三)——数据建模

根据前两次任务的数据探索结果,进行学习模型的训练,尝试多种模型,这些模型暂未调参,后面再补上

Y=data['status']
X=data.drop('status',axis=1)
# 把数据集拆分成训练集和测试集
seed = 7
test_size = 0.33
X_train, X_test, y_train, y_test = train_test_split(X, Y, test_size=test_size, random_state=seed)
from sklearn.preprocessing import StandardScaler
sc=StandardScaler()
sc.fit(X_test)
X_train_std=sc.transform(X_train)
X_test_std=sc.transform(X_test)
  1. 逻辑回归
    具体算法原理见前面的博文:
    https://blog.csdn.net/weixin_41940752/article/details/88952986
from sklearn.linear_model import LogisticRegression
#训练
model=LogisticRegression(C=1e5)
model.fit(X_train_std, y_train)
# 对测试集做预测
y_pred = model.predict(X_test_std)
  1. 决策树
    具体算法原理见前面博文:
    https://blog.csdn.net/weixin_41940752/article/details/88976566
from sklearn.tree import DecisionTreeClassifier
model=DecisionTreeClassifier()
model.fit(X_train_std, y_train)
# 对测试集做预测
y_pred = model.predict(X_test_std)
  1. SVM
    具体算法原理见前面博文:
    https://blog.csdn.net/weixin_41940752/article/details/90642533
from sklearn.svm import SVC
model=SVC()
model.fit(X_train_std, y_train)
# 对测试集做预测
y_pred = model.predict(X_test_std)
  1. 随机森林
    具体算法原理见前面博文:
    https://blog.csdn.net/weixin_41940752/article/details/98717868
from sklearn.ensemble import RandomForestClassifier
model=RandomForestClassifier()
model.fit(X_train_std, y_train)
# 对测试集做预测
y_pred = model.predict(X_test_std)
  1. Xgboost
    具体算法原理见前面博文:
    https://blog.csdn.net/weixin_41940752/article/details/96114516
from xgboost import XGBClassifier
# 拟合XGBoost模型
model = XGBClassifier()
model.fit(X_train, y_train)
# 对测试集做预测
y_pred = model.predict(X_test)

注:XGB算法能自动处理缺失值,具体原因我研究一下再补上

  1. GBDT
    具体算法原理见前面博文:
    https://blog.csdn.net/weixin_41940752/article/details/98948824
from sklearn.ensemble.gradient_boosting import GradientBoostingClassifier
model=GradientBoostingClassifier()
model.fit(X_train, y_train)
# 对测试集做预测
y_pred = model.predict(X_test)
  1. Adaboost
    具体算法原理见前面博文:
    https://blog.csdn.net/weixin_41940752/article/details/94021240
from sklearn.ensemble import AdaBoostClassifier
model=AdaBoostClassifier()
model.fit(X_train, y_train)
# 对测试集做预测
y_pred = model.predict(X_test)
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值