def ys_plot_model(X, y, model): plt.figure(figsize=(8,6)) bx = plt.subplot(111) # bx.set_ylabel('Intensity', )#fontsize = 16, # bx.set_xlabel('Diffraction angle 2$\Theta$ (deg.)', )#fontsize = 16 plt.axis([axl, axr, byl, byh]) model.cpu() X=X.cpu() mesh1 = np.arange(axl, axr, 0.01) mesh2 = np.arange(byl, byh, 0.01) xx, yy = np.meshgrid(mesh1, mesh2) with torch.no_grad(): data = torch.from_numpy(np.vstack((xx.reshape(-1), yy.reshape(-1))).T).float().cpu() Z = model(data).detach() Z = np.argmax(Z, axis=1).reshape(xx.shape) plt.contourf(xx, yy, Z, cmap=plt.cm.rainbow, alpha=0.3) yue_plot_data(X, y, axl,axr,byl,byh) 是什么意思
时间: 2023-06-04 09:02:13 浏览: 122
plot-number-train.rar_plot data_site:www.pudn.com_train_标号 matla
这段代码定义了一个函数 ys_plot_model,这个函数有三个输入参数:X、y 和 model。接着这个函数定义了一个图形对象,大小为 8*6,然后在这个图形对象上绘制一个子图,编号为 111,也就是只有一个子图。
阅读全文