算法实践(一)【任务1.1 - 模型构建】

【任务1.1 - 模型构建】
给定数据集,数据三七分,随机种子2018。(在任务1中什么都不用考虑,即不需数据处理和模型调参) 调用sklearn的包,简单构建逻辑回归、SVM和决策树3个模型,评分方式任意(e.g. 准确度和auc值)。

# -*- coding: utf-8 -*-
# 读取数据
import pandas as pd
data_all = pd.read_csv('data_all.csv',encoding='gbk')

# 划分数据集 (37分,随机种子2018)
from sklearn.model_selection import train_test_split
features = [x for x in data_all.columns if x not in ['status']]
x = data_all[features]
y = data_all['status']
x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=0.3,random_state=2018)

# 构建逻辑回归模型
from sklearn.linear_model import LogisticRegression
lr = LogisticRegression(random_state =2018)
lr.fit(x_train, y_train)

# 构建SVM模型
from sklearn.svm import LinearSVC
svm_linearSVC = LinearSVC(random_state=2018)
svm_linearSVC.fit(x_train, y_train)

# 构建决策树模型
from sklearn.tree import DecisionTreeClassifier
tree = DecisionTreeClassifier(random_state=2018)
tree.fit(x_train, y_train)

# 模型评价
lr_acc = lr.score(x_test, y_test)
svm_acc = svm_linearSVC.score(x_test, y_test)
tree_acc = tree.score(x_test, y_test)
print("LogisticRegression Acc: ", lr_acc)
print("SVM Acc: ", svm_acc)
print("DecisionTree Acc: ", tree_acc)

运行结果:

LogisticRegression Acc: 0.7484232655921513
SVM Acc: 0.7484232655921513
DecisionTree Acc: 0.6846531184302733
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值