机器学习过程中经常需要可视化,有助于加强对模型和参数的理解。 下面对梯度下降过程进行动图演示,可以修改不同的学习率,观看效果。 import numpy as np import matplotlib.pyplot as plt from IPython import display X = 2*np.random.rand(100,1) y = 4+3*X+np.random.randn(100,1) # randn正态分布 X_b = np.c_[np.ones((100,1)),X] # c_行数相等,左右拼接 eta = 0.1 # 学习率 n_iter = 1000 # 迭代次数