From 864ff3aca403aa36ef37a83258a62a3a89ee8827 Mon Sep 17 00:00:00 2001 From: Lukas Beiske Date: Fri, 20 Oct 2023 12:39:14 +0200 Subject: [PATCH 1/3] Add ctapipe-train-disp-reconstructor config to quickstart tool --- ctapipe/tools/quickstart.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/ctapipe/tools/quickstart.py b/ctapipe/tools/quickstart.py index 317d5723ab2..5244ce482f9 100644 --- a/ctapipe/tools/quickstart.py +++ b/ctapipe/tools/quickstart.py @@ -17,6 +17,7 @@ "ml_preprocessing_config.yaml", "train_energy_regressor.yaml", "train_particle_classifier.yaml", + "train_disp_reconstructor.yaml", ] README_TEXT = f""" @@ -55,13 +56,14 @@ ctapipe-process --help-all ``` -## ctapipe-train-energy-regressor / ctapipe-train-particle-classifier configs +## ctapipe-train-energy-regressor / ctapipe-train-particle-classifier / ctapipe-train-disp-reconstructor configs Included here are also base configurations for training machine learning (ML) -models for energy regression and gamma/hadron separation. +models for energy regression, gamma/hadron separation and disp origin reconstruction. - `train_energy_regressor.yaml`: configuration of energy regression model - `train_particle_classifier.yaml`: configuration of particle classification model +- `train_disp_reconstructor.yaml`: configuration of disp reconstruction models This file was generated using ctapipe version {VERSION} From 4cfc82a87ccc8238abd4a3512187fd83574c43be Mon Sep 17 00:00:00 2001 From: Lukas Beiske Date: Fri, 20 Oct 2023 12:51:31 +0200 Subject: [PATCH 2/3] Add changelog --- docs/changes/2414.bugfix.rst | 2 ++ 1 file changed, 2 insertions(+) create mode 100644 docs/changes/2414.bugfix.rst diff --git a/docs/changes/2414.bugfix.rst b/docs/changes/2414.bugfix.rst new file mode 100644 index 00000000000..0e6c4243c33 --- /dev/null +++ b/docs/changes/2414.bugfix.rst @@ -0,0 +1,2 @@ +Add the example config for ctapipe-train-disp-reconstructor +to the list of configs generated by ctapipe-quickstart. From 81cd459c668eadeff67f41a6192e79168827c14a Mon Sep 17 00:00:00 2001 From: Lukas Beiske Date: Fri, 20 Oct 2023 13:06:23 +0200 Subject: [PATCH 3/3] Add note about performance of provided ml configs --- ctapipe/tools/quickstart.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/ctapipe/tools/quickstart.py b/ctapipe/tools/quickstart.py index 5244ce482f9..319ccf316c1 100644 --- a/ctapipe/tools/quickstart.py +++ b/ctapipe/tools/quickstart.py @@ -60,6 +60,8 @@ Included here are also base configurations for training machine learning (ML) models for energy regression, gamma/hadron separation and disp origin reconstruction. +NOTE: As these files are used for unit tests, they are optimized for very fast training +and will not result in well performing models. - `train_energy_regressor.yaml`: configuration of energy regression model - `train_particle_classifier.yaml`: configuration of particle classification model