前面一个博客机器学习之路——logistic回归讲了logistic回归的理论知识,现在咱们来看一下logistic回归如何用python来实现,代码、数据参考《机器学习实战》。
首先看下我们要处理的数据,
我们要做的就是通过logistic回归的方法,拟合一条曲线将红色和绿色的点区分开来。话不多说,直接上代码,代码里我已经尽可能的加了注释,让大家更容易看懂:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
#上两行代码作用是可以在注释里写中文
import numpy as np
import matplotlib.pyplot as plt
#将测试数据导入
def loadDataSet ():
dataMat = []; labelMat = []#dataMat为X输入数据,labelMat为输出0