Allow visualize.plot_results to return matplotlib axis if desired

[bw4sz]

Issue Title

Request: visualize.plot_results should return matplotlib axis

Summary

Currently, the visualize.plot_results function in the DeepForest package generates plots but does not return the matplotlib axis object. If a user wants to wrap them into a subplot, its cumbersome.

Something like

# Create subplots
fig, axes = plt.subplots(1, 2, figsize=(10, 5))

# Plot zero_shot results into the first axis
axes[0] = visualize.plot_results(zero_shot, ax=ax1)

# Plot fine_tuned results into the second axis
axes[1] = visualize.plot_results(fine_tuned, ax=ax2)

# Display the subplots
plt.show()

Feature Request

I propose that visualize.plot_results be modified to return the matplotlib axis object. This change would allow users to more easily incorporate the plots into subplots for comparing multiple models.

This should be done with a 'axes=True' argument.

Benefits

Returning the axis object will provide users with greater flexibility in customizing their plots and integrating them into larger figures, thereby enhancing the usability of the DeepForest package for comparative analysis.

Labels

• good first issue
• enhancement

You can submit this as a new issue here.

DeepForest/src/deepforest/visualize.py

Line 419 in e5f3301

None

DeepForest/src/deepforest/visualize.py

Line 534 in e5f3301

plt.show()

written with copilot

[gadhvirushiraj]

Hello, I would like to contribute to this issue

[bw4sz]

Sounds great. Check out the contribution guidelines. For this issue we want to maintain current behavior, but add an argument to allow axes to be returned to be used within a matplotlib subplots loop.