folivo/thermal-conduction-ml
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

random changes

Browse Source
This commit is contained in:
Filippo Olivo
2025-10-27 10:23:13 +01:00
parent f49817ca1e
commit 6e90ef5393
7 changed files with 325 additions and 80 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
ThermalSolver/point_module.py
View File

@@ -15,15 +15,20 @@ def _plot_mesh(x, y, y_pred):
y_pred = y_pred[x[:, 0] != -1]
tria = Triangulation(pos[:, 2], pos[:, 3])
plt.figure(figsize=(12, 5))
plt.subplot(1, 2, 1)
plt.figure(figsize=(18, 5))
plt.subplot(1, 3, 1)
plt.tricontourf(tria, y.squeeze().numpy(), levels=14)
plt.colorbar()
plt.title("True temperature")
plt.subplot(1, 2, 2)
plt.subplot(1, 3, 2)
plt.tricontourf(tria, y_pred.squeeze().numpy(), levels=14)
plt.colorbar()
plt.title("Predicted temperature")
plt.subplot(1, 3, 3)
plt.tricontourf(tria, torch.abs(y_pred - y).squeeze().numpy(), levels=14)
plt.colorbar()
plt.title("Error")
plt.suptitle("PointNet", fontsize=16)
plt.savefig("point_net.png", dpi=300)