Skip to content

Commit

Permalink
Create a new method called fit to run all processes after initializat…
Browse files Browse the repository at this point in the history
…ion of MetaModel:

prepare_data_and_computing_op, build_stacking_learning, event_extraction_optimization, and plot_save
  • Loading branch information
menouarazib committed Sep 15, 2023
1 parent 7e1ef56 commit c5ce2d3
Showing 1 changed file with 12 additions and 0 deletions.
12 changes: 12 additions & 0 deletions eventdetector_ts/metamodel/meta_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -438,3 +438,15 @@ def plot_save(self, show_plots: bool = True) -> None:
self.plotter.plot_prediction()
self.plotter.plot_predicted_events()
self.plotter.plot_delta_t(bins=10)

def fit(self) -> None:
"""
Run prepare_data_and_computing_op, build_stacking_learning, event_extraction_optimization, and plot_save
Returns:
None
"""
self.prepare_data_and_computing_op()
self.build_stacking_learning()
self.event_extraction_optimization()
self.plot_save()

0 comments on commit c5ce2d3

Please sign in to comment.