Add option to generate the average model when averaging FOOOFs

Author: TomDonoghue

fooof/funcs.py

@@ -10,7 +10,7 @@
 ###################################################################################################
 ###################################################################################################
 
-def average_fg(fg, bands, avg_method='mean'):
+def average_fg(fg, bands, avg_method='mean', generate_model=True):
     """Average across a FOOOFGroup object.
 
     Parameters
@@ -21,6 +21,8 @@ def average_fg(fg, bands, avg_method='mean'):
         Bands object that defines the frequency bands to collapse peaks across.
     avg : {'mean', 'median'}
         Averaging function to use.
+    generate_model : bool, optional, default: True
+        Whether to generate the model for the averaged parameters.
 
     Returns
     -------
@@ -56,6 +58,10 @@ def average_fg(fg, bands, avg_method='mean'):
     fm.add_meta_data(fg.get_meta_data())
     fm.add_results(results)
 
+    # Generate the model from the average parameters
+    if generate_model:
+        fm._regenerate_model()
+
     return fm