逻辑回归-6.解决多分类问题

逻辑回归是使用回归的方式来解决分类问题。以前说过,逻辑回归只能解决二分类问题,为了解决多分类问题,能够使用OVR和OVO方法
算法

  • OVR(One Vs Rest)
    某个分类算法有N类,将某一类和剩余的类比较做为二分类问题,N个类别进行N次分类,获得N个二分类模型,给定一个新的样本点,求出每种二分类对应的几率,几率最高的一类做为新样本的预测结果。
    dom

  • OVO(One Vs One)
    某个分类算法有N类,将某一类和另外一类比较做为二分类问题,总共可分为\(C^2_n\)种不一样的二分类模型,给定一个新的样本点,求出每种二分类对应的几率,几率最高的一类做为新样本的预测结果。
    spa

加载鸢尾花数据集(数据集有三类结果):code

import numpy
from sklearn import datasets
from sklearn.model_selection import train_test_split

iris = datasets.load_iris()
# 为了数据可视化,只取数据集的前两个特征
x = iris.data[:,:2]
y = iris.target
x_train,x_test,y_train,y_test = train_test_split(x,y,random_state=666)

scikit-learn中默认支持多分类,且多分类方法默认为OVRblog

from sklearn.linear_model import LogisticRegression

log_reg = LogisticRegression()
log_reg.fit(x_train,y_train)

画出决策边界:
ci

使用OVO多分类方法:get

log_reg2 = LogisticRegression(multi_class='multinomial',solver='newton-cg')
log_reg2.fit(x_train,y_train)

scikit-learn中的OVR和OVO类¶

from sklearn.multiclass import OneVsRestClassifier,OneVsOneClassifier

# 使数据全部的特征值参与运算
x = iris.data
y = iris.target
x_train,x_test,y_train,y_test = train_test_split(x,y,random_state=666)
  • OVR
log_reg1 = LogisticRegression()
OVR = OneVsRestClassifier(log_reg1)
OVR.fit(x_train,y_train)

准确率:
it

  • OVO
log_reg2 = LogisticRegression()
OVR = OneVsRestClassifier(log_reg2)
OVR.fit(x_train,y_train)

准确率:
io

相关文章
相关标签/搜索