folivo/PINA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Plotter v0.1 update (#211)

Browse Source 
* Commented a line responsible for bugs. No labels attribute in torch.Tensor object
* codacy

---------

Co-authored-by: Dario Coscia <93731561+dario-coscia@users.noreply.github.com>
This commit is contained in:
Giovanni_Canali
2023-11-15 15:45:01 +01:00
committed by Nicola Demo
parent 0b7a307cf1
commit 9bf057e2c6
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
pina/plotter.py
View File

@@ -88,7 +88,10 @@ class Plotter:
truth_output = truth_solution(pts).float() truth_output = truth_solution(pts).float()
ax.plot(pts, truth_output.detach(), label='True solution', **kwargs) ax.plot(pts, truth_output.detach(), label='True solution', **kwargs)
plt.ylabel(pred.labels[0]) # TODO: pred is a torch.Tensor, so no labels is available
# extra variable for labels should be
# passed in the function arguments.
# plt.ylabel(pred.labels[0])
plt.legend() plt.legend()
plt.show() plt.show()