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:
committed by
Nicola Demo
parent
0b7a307cf1
commit
9bf057e2c6
@@ -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()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user