Plotter v0.1 update (#211)

* 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

View File

@@ -88,7 +88,10 @@ class Plotter:
truth_output = truth_solution(pts).float()
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.show()