吴恩达机器学习线性回归作业

一、单变量线性回归：

导入相关库：

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt

 读取csv数据文件并查看数据集的前五行

f = pd.read_csv('work/ex1data1.txt', names=['Population', 'Profit'])
f.head()

 

 可视化该数据集

f.plot(kind='scatter', x='Population', y='Profit')
plt.show()

 

损失函数计算：

def computeCost(X, Y, theta):
    cost = np.power((X * theta.T - Y), 2)
    return np.sum(cost) / (2 * len(X))

在f的第一列插入常数1