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

Update plotter.py

Browse Source 
* minor update for 1d plots
This commit is contained in:
Dario Coscia
2023-11-16 17:47:26 +01:00
committed by Nicola Demo
parent 9bf057e2c6
commit 48b2e339b5
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
pina/plotter.py
View File

@@ -197,7 +197,7 @@ class Plotter:
truth_solution = getattr(solver.problem, 'truth_solution', None) truth_solution = getattr(solver.problem, 'truth_solution', None)
if len(v) == 1: if len(v) == 1:
self._1d_plot(pts, predicted_output, method, truth_solution, self._1d_plot(pts.extract(v), predicted_output, method, truth_solution,
**kwargs) **kwargs)
elif len(v) == 2: elif len(v) == 2:
self._2d_plot(pts, predicted_output, v, res, method, truth_solution, self._2d_plot(pts, predicted_output, v, res, method, truth_solution,
@@ -208,7 +208,6 @@ class Plotter:
plt.savefig(filename) plt.savefig(filename)
else: else:
plt.show() plt.show()
plt.close()
def plot_loss(self, def plot_loss(self,
trainer, trainer,