本文是吴恩达机器学习课程的第一次编程作业:Linear Regression 的基础作业,即影响因子是单个变量的情况,用python实现。
本作业包含5个文件,分别是:
ex1.py :程序的主入口
plot_data.py :可视化训练集
compute_cost.py :计算代价函数
gradient_descent.py :梯度向下算法
ex1data1.txt :训练集
作业文件和训练集数据下载地址:https://github.com/toanoyx/MachineLearning-AndrewNg-coursera-python/tree/master/ex1%20Linear%20Regression/ex1
下文是文件的源代码:
ex1.py (程序的主入口)
from matplotlib.colors import LogNorm
from mpl_toolkits.mplot3d import axes3d, Axes3D
from gradient_descent import *
from plot_data import *
'''第1部分 可视化训练集'''
print('Plotting Data...')
data = np.loadtxt('ex1data1.txt', de