在上一篇 线性回归参数估计–最小二乘法与梯度下降法完成算法学习后,在这篇博客中贴出Python实现的代码。
这里使用的数据集来自Computer Hardware Data Set 。
数据集下载地址
实现代码
#!/usr/bin/python3
# -*- coding: utf-8 -*-
import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
class linear_classifier(object):
"""docstring for linear_classifier"""
def __init__(self, train_set, label_set):
self.__train_set = train_set
self.__target_mat = np.mat(label_set).T
self.__thet = None
self.__errores = []