plot画图等高线contour 与 contourf 及API

contour 与 contourf  api

https://matplotlib.org/3.1.0/api/_as_gen/matplotlib.pyplot.contourf.html

 

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
 
 
# 计算x,y坐标对应的高度值
def f(x, y):
 return (1-x/2+x**3+y**5) * np.exp(-x**2-y**2)
 
# 生成x,y的数据
n = 256
x = np.linspace(-3, 3, n)
y = np.linspace(-3, 3, n)
 
# 把x,y数据生成mesh网格状的数据,因为等高线的显示是在网格的基础上添加上高度值
X, Y = np.meshgrid(x, y)
 
# 填充等高线
plt.contour(X, Y, f(X, Y))
# 显示图表
plt.show()
 

#result  

 

参考

https://blog.csdn.net/lens___/article/details/83960810

 

这段代码看起来是一个用于可视化神经网络训练过程中损失函数变化的工具。以下是代码的完整注释和说明: ``` # project states onto the main directions of the gradient updates using n samples over all steps starting from sample x # the directions are calculated using the last sample as a reference directions, state_ids, loss_coordinates = get_state_directions(states, n_states=10, start_from=0, reference_id=-1) # compute the losses over the main directions of the gradient updates x, y, Z, _ = get_loss_grid(net, data_loader, loss_fn, directions=directions, resolution=(20, 20), scale=loss_coordinates.abs().max().item()) # plot the landscape as a contour plot fig = plot_contour(np.copy(x), np.copy(y), np.copy(Z), scale=True) fig.add_traces(go.Scatter(x=np.copy(loss_coordinates[0].cpu().numpy()), y=np.copy(loss_coordinates[1].cpu().numpy()))) # print the losses for the selected states print('loss samples:', np.array(losses)[state_ids]) # show the plot using Plotly conf_pltly() init_notebook_mode(connected=False) iplot(fig) ``` 具体来说,代码的功能如下: 1. 使用 `get_state_directions` 函数将神经网络训练过程中的状态向量投影到梯度更新的方向上,得到主要的梯度更新方向。其中,`n_states` 参数指定了采样的状态数,`start_from` 参数指定了从哪个状态开始采样,`reference_id` 参数指定了用哪个状态作为参考状态计算梯度更新方向。 2. 使用 `get_loss_grid` 函数计算在主要的梯度更新方向上,神经网络对数据集的损失函数值。其中,`net` 参数指定了神经网络模型,`data_loader` 参数指定了数据集的数据加载器,`loss_fn` 参数指定了损失函数,`directions` 参数指定了梯度更新方向,`resolution` 参数指定了计算损失函数的网格分辨率,`scale` 参数指定了损失函数值的缩放因子。 3. 使用 `plot_contour` 函数将损失函数的值绘制成等高线图。同时,使用 `add_traces` 函数将采样点的坐标绘制在图上。 4. 使用 `print` 函数输出选定状态的损失函数值。 5. 使用 Plotly 的 `conf_pltly` 函数和 `init_notebook_mode` 函数配置绘图环境,然后使用 `iplot` 函数显示图像。 需要注意的是,这段代码中使用了一些自定义的函数,如 `get_state_directions`、`get_loss_grid`、`plot_contour` 等,需要先定义或导入这些函数才能运行此代码。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值