Commit 517612c

authored

Two changes in the plot_result() method.

1) Rename the parameter to `save_dir`. If it is not None, then save. 2) Keep the .show() function called regardless of saving the figure or not.

1 parent ccfe679 commit 517612cCopy full SHA for 517612c

File tree

1 file changed

-7

lines changed

pygad.py

1 file changed

-7

lines changed

`‎pygad.py`

Lines changed: 7 additions & 7 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, save_as=None):`
	`2533`	`+ def plot_result(self, title="PyGAD - Iteration vs. Fitness", xlabel="Generation", ylabel="Fitness", linewidth=3, save_dir=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,7 +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`	`+ save_dir: Directory to save the figure.`
`2544`	`2544`
`2545`	`2545`	`Returns the figure.`
`2546`	`2546`	`"""`
`@@ -2557,10 +2557,10 @@ def plot_result(self, title="PyGAD - Iteration vs. Fitness", xlabel="Generation"`
`2557`	`2557`	`matplotlib.pyplot.xlabel(xlabel)`
`2558`	`2558`	`matplotlib.pyplot.ylabel(ylabel)`
`2559`	`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`
	`2560`	`+ if notsave_dir is None:`
	`2561`	`+ matplotlib.pyplot.savefig(fname=save_dir,`
	`2562`	`+ bbox_inches='tight')`
	`2563`	`+ matplotlib.pyplot.show()`
`2564`	`2564`
`2565`	`2565`	`return fig`
`2566`	`2566`
`@@ -2589,4 +2589,4 @@ def load(filename):`
`2589`	`2589`	`raise FileNotFoundError("Error reading the file {filename}. Please check your inputs.".format(filename=filename))`
`2590`	`2590`	`except:`
`2591`	`2591`	`raise BaseException("Error loading the file. Please check if the file exists.")`
`2592`		`- return ga_in`
	`2592`	`+ return ga_in`

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 517612c

File tree

1 file changed

1 file changed

`‎pygad.py`

0 commit comments