Merge pull request #818 from rasbt/learning-curve

learning curve doc

Author: rasbt

docs/sources/CHANGELOG.md

@@ -25,6 +25,7 @@ The CHANGELOG for the current development version is available at
 - Updates unit tests for scikit-learn 0.24.1 compatibility. ([#774](https://github.com/rasbt/mlxtend/pull/774))
 - `StackingRegressor` now requires setting `StackingRegressor(..., multi_output=True)` if the target is multi-dimensional; this allows for better input validation. ([#802](https://github.com/rasbt/mlxtend/pull/802))
 - Removes deprecated `res` argument from `plot_decision_regions`. ([#803](https://github.com/rasbt/mlxtend/pull/803))
+- Adds a `title_fontsize` parameter to `plot_learning_curves` for controlling the title font size; also the plot style is now the matplotlib default. ([#818](https://github.com/rasbt/mlxtend/pull/818))
 
 ##### Bug Fixes
 

docs/sources/user_guide/plotting/plot_learning_curves.ipynb

@@ -16,8 +16,10 @@ def plot_learning_curves(X_train, y_train,
                          train_marker='o',
                          test_marker='^',
                          scoring='misclassification error',
-                         suppress_plot=False, print_model=True,
-                         style='fivethirtyeight',
+                         suppress_plot=False,
+                         print_model=True,
+                         title_fontsize=12,
+                         style='default',
                          legend_loc='best'):
     """Plots learning curves of a classifier.
 
@@ -49,8 +51,11 @@ def plot_learning_curves(X_train, y_train,
         for testing purposes.
     print_model : bool (default: True)
         Print model parameters in plot title if True.
-    style : str (default: 'fivethirtyeight')
-        Matplotlib style
+    title_fontsize : int (default: 12)
+        Determines the size of the plot title font.
+    style : str (default: 'default')
+        Matplotlib style. For more styles, please see
+        https://matplotlib.org/stable/gallery/style_sheets/style_sheets_reference.html
     legend_loc : str (default: 'best')
         Where to place the plot legend:
         {'best', 'upper left', 'upper right', 'lower left', 'lower right'}
@@ -124,7 +129,9 @@ def misclf_err(y_predict, y):
         with plt.style.context(style):
             plt.ylabel('Performance ({})'.format(scoring))
             if print_model:
-                plt.title('Learning Curves\n\n{}\n'.format(model))
+                plt.title(
+                    'Learning Curves\n\n{}\n'.format(model),
+                    fontsize=title_fontsize)
             plt.legend(loc=legend_loc, numpoints=1)
             plt.xlim([0, 110])
             max_y = max(max(test_errors), max(training_errors))