上一次,写了MGD、SGD、MBGD的代码实现,现在,我们来康康实例
我们以大名鼎鼎的鸢尾花数据集为例:
https://archive.ics.uci.edu/ml/machine-learning-databases/iris/
下载这个iris.data即可
将其置于当前工作文件夹即可
先导入需要的库:
import numpy as np
import pandas as pd
import random
然后将我们上一次写的函数copy过来:
def MGD_train(X, y, alpha=0.0001, maxIter=1000, theta_old=None):
1