Skip to content

Commit

Permalink
Add docstring to helper function
Browse files Browse the repository at this point in the history
  • Loading branch information
ArturoAmorQ committed May 7, 2024
1 parent 2e86f09 commit 99bed3b
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion python_scripts/ensemble_gradient_boosting.py
Original file line number Diff line number Diff line change
Expand Up @@ -86,7 +86,8 @@ def generate_data(n_samples=50):

# %%
def plot_decision_tree_with_residuals(y_train, y_train_pred, y_test_pred):
# Create a plot and get the Axes object
"""Plot the synthetic data, predictions, and residuals for a decision tree.
Handles are used to create custom legends for the plot."""
fig, ax = plt.subplots()
# plot the data
sns.scatterplot(
Expand Down

0 comments on commit 99bed3b

Please sign in to comment.