集成学习作业Task01

本文为Datawhale组队学习项目——集成学习的作业Task

相关视频
相关资料

作业1

第一问：

通过代码验证$a$对于图像的形状没有影响，$b$则会根据$b<0、b=0、b>0$三种情况，出现三种图像。如下图所示。

b>0	b=0	b<0

作业2,采用牛顿法计算极值点，迭代500次，步长取0.1，代码：

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import time
%matplotlib inline
from mpl_toolkits.mplot3d import Axes3D

##Rosenbrock函数
class Rosenbrock():
    def __init__(self,a,b):
        #确定数值范围
        self.x1 = np.arange(-100, 100, 0.0001)
        self.x2 = np.arange(-100, 100, 0.0001)
        #self.x1, self.x2 = np.meshgrid(self.x1, self.x2)
        self.a = a
        self.b = b
        
        self.newton_times = 1000
        self.answers = []
        self.min_answer_z = []


    # 准备数据
    def data(self):
        z = np.square(self.a - self.x1) + self.b * np.square(self.x2 - np.square(self.x1))
        #print(z.shape)
        return z

    # 随机牛顿
    def snt(self,x1,x2,z,alpha):
        #取出随机点
        rand_init = np.random.randint(0,z.shape[0])
        x1_init,x2_init,z_init = x1[rand_init],x2[rand_init],z[rand_init]
        x_0 =np.array([x1_init,x2_init]).reshape((-1,1))
        #print(x_0)

        
        #迭代求解
        for i in range(self.newton_times):
            x_i = x_0 - alpha*np.matmul(np.linalg.inv(np.array([[(12*x1_init**2-4*x2_init)*self.b-2*self.a,-4*x1_init*self.b],
                                                          [-4*x1_init*self.b,2*self.b]])),
                                  np.array([(4*x1_init**3-4*x1_init*x2_init+2*x1_init)*self.b-2*self.a,
                                            (-2*x1_init**2+2*x2_init)*self.b]).reshape((-1,1)))
            x_0 = x_i
            x1_init = x_0[0,0]
            x2_init = x_0[1,0]
        answer = x_0
        return answer


    # 绘图
    def plot_data(self,min_x1,min_x2,min_z):
        x1 = np.arange(-100, 100, 0.1)
        x2 = np.arange(-100, 100, 0.1)
        x1, x2 = np.meshgrid(x1, x2)
        a = self.a
        b = self.b
        z = np.square(a - x1) + b * np.square(x2 - np.square(x1))
        fig4 = plt.figure()
        ax4 = plt.axes(projection='3d')
        ax4.plot_surface(x1, x2, z, alpha=0.3, cmap='winter')  # 生成表面， alpha 用于控制透明度
        ax4.contour(x1, x2, z, zdir='z', offset=-3, cmap="rainbow")  # 生成z方向投影，投到x-y平面
        ax4.contour(x1, x2, z, zdir='x', offset=-6, cmap="rainbow")  # 生成x方向投影，投到y-z平面
        ax4.contour(x1, x2, z, zdir='y', offset=6, cmap="rainbow")  # 生成y方向投影，投到x-z平面
        ax4.contourf(x1, x2, z, zdir='y', offset=6, cmap="rainbow")  # 生成y方向投影填充，投到x-z平面，contourf()函数
        ax4.scatter(min_x1,min_x2,min_z,c='r')
        # 设定显示范围
        ax4.set_xlabel('X')
        ax4.set_ylabel('Y')
        ax4.set_zlabel('Z')
        plt.show()

    # 开始
    def start(self):
#         times = int(input("请输入需要随机优化的次数："))
        times=500
#         alpha = float(input("请输入随机优化的步长"))
        alpha=0.1
        z = self.data()
        start_time = time.time()
        for i in range(times):
            answer = self.snt(self.x1,self.x2,z,alpha)
            self.answers.append(answer)
        min_answer = np.array(self.answers)
        for i in range(times):
            self.min_answer_z.append((1-min_answer[i,0,0])**2+(min_answer[i,1,0]-min_answer[i,0,0]**2)**2)
        optimal_z = np.min(np.array(self.min_answer_z))
        optimal_z_index = np.argmin(np.array(self.min_answer_z))
        optimal_x1,optimal_x2 = min_answer[optimal_z_index,0,0],min_answer[optimal_z_index,1,0]
        print("极值点,x1=",optimal_x1,"x2=",optimal_x2)
        end_time = time.time()
        running_time = end_time-start_time
        print("优化的时间:%.2f秒!" % running_time)
        self.plot_data(optimal_x1,optimal_x2,optimal_z)

取a=1,b=1和a=-1,b=1,分别计算极值点，

if __name__ == '__main__':
    snt = Rosenbrock(1,1)
    snt.start()
    
    snt = Rosenbrock(-1,1)
    snt.start()

得到结果