|
869 | 869 | "output = net(input_data).detach()\n", |
870 | 870 | "\n", |
871 | 871 | "# visualize data\n", |
872 | | - "#fig, axes = plt.subplots(nrows=1, ncols=2, figsize=(8, 3))\n", |
873 | | - "#pic1 = axes[0].scatter(grid[:, 0], grid[:, 1], c=input_data[0, 0, :, -1])\n", |
874 | | - "#axes[0].set_title(\"Real\")\n", |
875 | | - "#fig.colorbar(pic1)\n", |
876 | | - "#plt.subplot(1, 2, 2)\n", |
877 | | - "#pic2 = axes[1].scatter(grid[:, 0], grid[:, 1], c=output[0, 0, :, -1])\n", |
878 | | - "#axes[1].set_title(\"Autoencoder\")\n", |
879 | | - "#fig.colorbar(pic2)\n", |
880 | | - "#plt.tight_layout()\n", |
881 | | - "#plt.show()\n" |
| 872 | + "fig, axes = plt.subplots(nrows=1, ncols=2, figsize=(8, 3))\n", |
| 873 | + "pic1 = axes[0].scatter(grid[:, 0], grid[:, 1], c=input_data[0, 0, :, -1])\n", |
| 874 | + "axes[0].set_title(\"Real\")\n", |
| 875 | + "fig.colorbar(pic1)\n", |
| 876 | + "plt.subplot(1, 2, 2)\n", |
| 877 | + "pic2 = axes[1].scatter(grid[:, 0], grid[:, 1], c=output[0, 0, :, -1])\n", |
| 878 | + "axes[1].set_title(\"Autoencoder\")\n", |
| 879 | + "fig.colorbar(pic2)\n", |
| 880 | + "plt.tight_layout()\n", |
| 881 | + "plt.show()\n" |
882 | 882 | ] |
883 | 883 | }, |
884 | 884 | { |
|
963 | 963 | "output = net.decoder(latent, input_data2).detach()\n", |
964 | 964 | "\n", |
965 | 965 | "# show the picture\n", |
966 | | - "#fig, axes = plt.subplots(nrows=1, ncols=2, figsize=(8, 3))\n", |
967 | | - "#pic1 = axes[0].scatter(grid2[:, 0], grid2[:, 1], c=input_data2[0, 0, :, -1])\n", |
968 | | - "#axes[0].set_title(\"Real\")\n", |
969 | | - "#fig.colorbar(pic1)\n", |
970 | | - "#plt.subplot(1, 2, 2)\n", |
971 | | - "#pic2 = axes[1].scatter(grid2[:, 0], grid2[:, 1], c=output[0, 0, :, -1])\n", |
972 | | - "# axes[1].set_title(\"Up-sampling\")\n", |
973 | | - "#fig.colorbar(pic2)\n", |
974 | | - "#plt.tight_layout()\n", |
975 | | - "#plt.show()\n" |
| 966 | + "fig, axes = plt.subplots(nrows=1, ncols=2, figsize=(8, 3))\n", |
| 967 | + "pic1 = axes[0].scatter(grid2[:, 0], grid2[:, 1], c=input_data2[0, 0, :, -1])\n", |
| 968 | + "axes[0].set_title(\"Real\")\n", |
| 969 | + "fig.colorbar(pic1)\n", |
| 970 | + "plt.subplot(1, 2, 2)\n", |
| 971 | + "pic2 = axes[1].scatter(grid2[:, 0], grid2[:, 1], c=output[0, 0, :, -1])\n", |
| 972 | + "axes[1].set_title(\"Up-sampling\")\n", |
| 973 | + "fig.colorbar(pic2)\n", |
| 974 | + "plt.tight_layout()\n", |
| 975 | + "plt.show()\n" |
976 | 976 | ] |
977 | 977 | }, |
978 | 978 | { |
|
1051 | 1051 | "output = net.decoder(latent, input_data2).detach()\n", |
1052 | 1052 | "\n", |
1053 | 1053 | "# show the picture\n", |
1054 | | - "#fig, axes = plt.subplots(nrows=1, ncols=2, figsize=(8, 3))\n", |
1055 | | - "#pic1 = axes[0].scatter(grid2[:, 0], grid2[:, 1], c=input_data2[0, 0, :, -1])\n", |
1056 | | - "#axes[0].set_title(\"Real\")\n", |
1057 | | - "#fig.colorbar(pic1)\n", |
1058 | | - "#plt.subplot(1, 2, 2)\n", |
1059 | | - "#pic2 = axes[1].scatter(grid2[:, 0], grid2[:, 1], c=output[0, 0, :, -1])\n", |
1060 | | - "#axes[1].set_title(\"Autoencoder not re-trained\")\n", |
1061 | | - "#fig.colorbar(pic2)\n", |
1062 | | - "#plt.tight_layout()\n", |
1063 | | - "#plt.show()\n", |
| 1054 | + "fig, axes = plt.subplots(nrows=1, ncols=2, figsize=(8, 3))\n", |
| 1055 | + "pic1 = axes[0].scatter(grid2[:, 0], grid2[:, 1], c=input_data2[0, 0, :, -1])\n", |
| 1056 | + "axes[0].set_title(\"Real\")\n", |
| 1057 | + "fig.colorbar(pic1)\n", |
| 1058 | + "plt.subplot(1, 2, 2)\n", |
| 1059 | + "pic2 = axes[1].scatter(grid2[:, 0], grid2[:, 1], c=output[0, 0, :, -1])\n", |
| 1060 | + "axes[1].set_title(\"Autoencoder not re-trained\")\n", |
| 1061 | + "fig.colorbar(pic2)\n", |
| 1062 | + "plt.tight_layout()\n", |
| 1063 | + "plt.show()\n", |
1064 | 1064 | "\n", |
1065 | 1065 | "# calculate l2 error\n", |
1066 | 1066 | "print(f'l2 error: {l2_error(input_data2[0, 0, :, -1], output[0, 0, :, -1]):.2%}')" |
|
0 commit comments