Commit c2e78b2

authored

Merge pull request #47 from rengel8/save-matplot-option

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.

2 parents 19c20c4 + 517612c commit c2e78b2Copy full SHA for c2e78b2

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
`@@ -2792,7 +2792,7 @@ def best_solution(self, pop_fitness=None):`
`2792`	`2792`
`2793`	`2793`	`return best_solution, best_solution_fitness, best_match_idx`
`2794`	`2794`
`2795`		`- def plot_result(self, title="PyGAD - Iteration vs. Fitness", xlabel="Generation", ylabel="Fitness", linewidth=3):`
	`2795`	`+ def plot_result(self, title="PyGAD - Iteration vs. Fitness", xlabel="Generation", ylabel="Fitness", linewidth=3, save_dir=None):`
`2796`	`2796`
`2797`	`2797`	`"""`
`2798`	`2798`	`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.`
`@@ -2802,6 +2802,7 @@ def plot_result(self, title="PyGAD - Iteration vs. Fitness", xlabel="Generation"`
`2802`	`2802`	`xlabel: Label on the X-axis.`
`2803`	`2803`	`ylabel: Label on the Y-axis.`
`2804`	`2804`	`linewidth: Line width of the plot.`
	`2805`	`+ save_dir: Directory to save the figure.`
`2805`	`2806`
`2806`	`2807`	`Returns the figure.`
`2807`	`2808`	`"""`
`@@ -2817,7 +2818,12 @@ def plot_result(self, title="PyGAD - Iteration vs. Fitness", xlabel="Generation"`
`2817`	`2818`	`matplotlib.pyplot.title(title)`
`2818`	`2819`	`matplotlib.pyplot.xlabel(xlabel)`
`2819`	`2820`	`matplotlib.pyplot.ylabel(ylabel)`
	`2821`	`+`
	`2822`	`+ if not save_dir is None:`
	`2823`	`+ matplotlib.pyplot.savefig(fname=save_dir,`
	`2824`	`+ bbox_inches='tight')`
`2820`	`2825`	`matplotlib.pyplot.show()`
	`2826`	`+`
`2821`	`2827`	`return fig`
`2822`	`2828`
`2823`	`2829`	`def save(self, filename):`
`@@ -2845,4 +2851,4 @@ def load(filename):`
`2845`	`2851`	`raise FileNotFoundError("Error reading the file {filename}. Please check your inputs.".format(filename=filename))`
`2846`	`2852`	`except:`
`2847`	`2853`	`raise BaseException("Error loading the file. Please check if the file exists.")`
`2848`		`- return ga_in`
	`2854`	`+ 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 c2e78b2

File tree

1 file changed

1 file changed

`‎pygad.py`

0 commit comments