diff --git a/docs/source/examples/plot_gradient.py b/docs/source/examples/plot_gradient.py index 38fa225e..190d8b29 100644 --- a/docs/source/examples/plot_gradient.py +++ b/docs/source/examples/plot_gradient.py @@ -69,9 +69,7 @@ score_display = ScoreScatterDisplay.from_estimator( cca, [X_train, Y_train], [X_test, Y_test] ) -score_display.plot( - title=f"CCA (Time: {elapsed_time:.2f} s)" -) +score_display.plot(title=f"CCA (Time: {elapsed_time:.2f} s)") plt.show() # %% @@ -102,7 +100,9 @@ # We plot the transformed views on a scatter plot with different colors for train and test sets # Use ScoreScatterDisplay or a similar plotting class for the visualization - score_display = ScoreScatterDisplay.from_estimator(ccaey, [X_train, Y_train], [X_test, Y_test]) + score_display = ScoreScatterDisplay.from_estimator( + ccaey, [X_train, Y_train], [X_test, Y_test] + ) score_display.plot( title=f"CCA_EY (Batch size: {batch_size}, Time: {elapsed_time:.2f} s)" )