Commit ccfe679

committed

Update pygad.py

Allows to save the plot into a file as an option, since linux might throw an error showing it.

1 parent dd7a670 commit ccfe679Copy full SHA for ccfe679

File tree

1 file changed

-2

lines changed

pygad.py

1 file changed

-2

lines changed

`‎pygad.py`

Lines changed: 8 additions & 2 deletions

Original file line number	Diff line number	Diff line change
`@@ -2530,7 +2530,7 @@ def best_solution(self, pop_fitness=None):`
`2530`	`2530`
`2531`	`2531`	`return best_solution, best_solution_fitness, best_match_idx`
`2532`	`2532`
`2533`		`- def plot_result(self, title="PyGAD - Iteration vs. Fitness", xlabel="Generation", ylabel="Fitness", linewidth=3):`
	`2533`	`+ def plot_result(self, title="PyGAD - Iteration vs. Fitness", xlabel="Generation", ylabel="Fitness", linewidth=3, save_as=None):`
`2534`	`2534`
`2535`	`2535`	`"""`
`2536`	`2536`	`Creates and shows a plot that summarizes how the fitness value evolved by generation. Can only be called after completing at least 1 generation. If no generation is completed, an exception is raised.`
`@@ -2540,6 +2540,7 @@ def plot_result(self, title="PyGAD - Iteration vs. Fitness", xlabel="Generation"`
`2540`	`2540`	`xlabel: Label on the X-axis.`
`2541`	`2541`	`ylabel: Label on the Y-axis.`
`2542`	`2542`	`linewidth: Line width of the plot.`
	`2543`	`+ save_as: save image instead of showing it.`
`2543`	`2544`
`2544`	`2545`	`Returns the figure.`
`2545`	`2546`	`"""`
`@@ -2555,7 +2556,12 @@ def plot_result(self, title="PyGAD - Iteration vs. Fitness", xlabel="Generation"`
`2555`	`2556`	`matplotlib.pyplot.title(title)`
`2556`	`2557`	`matplotlib.pyplot.xlabel(xlabel)`
`2557`	`2558`	`matplotlib.pyplot.ylabel(ylabel)`
`2558`		`- matplotlib.pyplot.show()`
	`2559`	`+`
	`2560`	`+ if save_as is None:`
	`2561`	`+ matplotlib.pyplot.show() # show plot`
	`2562`	`+ else:`
	`2563`	`+ fig.savefig(f'{save_as}.png') # save as image`
	`2564`	`+`
`2559`	`2565`	`return fig`
`2560`	`2566`
`2561`	`2567`	`def save(self, filename):`

0 commit comments

Comments

(0)

Navigation Menu

Search code, repositories, users, issues, pull requests...

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Uh oh!

Commit ccfe679

File tree

1 file changed

1 file changed

`‎pygad.py`

0 commit comments