@@ -34,15 +34,17 @@ def configure_training(analysis_type):
3434 parser .add_argument (
3535 "--model_prefix" ,
3636 required = True ,
37- help = ("Path to directory for writing XGBoost models (without n_tel / energy bin suffix)." ),
37+ help = (
38+ "Path prefix for writing XGBoost models (without energy bin suffix; "
39+ "suffix is added automatically)."
40+ ),
3841 )
3942 parser .add_argument (
4043 "--hyperparameter_config" ,
4144 help = "Path to JSON file with hyperparameter configuration." ,
4245 default = None ,
4346 type = str ,
4447 )
45- parser .add_argument ("--n_tel" , type = int , help = "Telescope multiplicity (2, 3, or 4)." )
4648 parser .add_argument (
4749 "--train_test_fraction" ,
4850 type = float ,
@@ -111,7 +113,6 @@ def configure_training(analysis_type):
111113
112114 _logger .info (f"--- XGBoost { analysis_type } training ---" )
113115 _logger .info (f"Observatory: { model_configs .get ('observatory' )} " )
114- _logger .info (f"Telescope multiplicity: { model_configs .get ('n_tel' )} " )
115116 _logger .info (f"Model output prefix: { model_configs .get ('model_prefix' )} " )
116117 _logger .info (f"Train vs test fraction: { model_configs ['train_test_fraction' ]} " )
117118 _logger .info (f"Random state: { model_configs ['random_state' ]} " )
@@ -131,15 +132,14 @@ def configure_training(analysis_type):
131132
132133 if analysis_type == "stereo_analysis" :
133134 model_configs ["pre_cuts" ] = pre_cuts_regression (
134- model_configs . get ( "n_tel" ), min_images = model_configs .get ("min_images" , 2 )
135+ min_images = model_configs .get ("min_images" , 2 )
135136 )
136137 elif analysis_type == "classification" :
137138 _logger .info (f"Energy bin { model_configs ['energy_bin_number' ]} " )
138139 model_parameters = utils .load_model_parameters (
139140 model_configs ["model_parameters" ], model_configs ["energy_bin_number" ]
140141 )
141142 model_configs ["pre_cuts" ] = pre_cuts_classification (
142- model_configs .get ("n_tel" ),
143143 e_min = np .power (10.0 , model_parameters .get ("energy_bins_log10_tev" , []).get ("E_min" )),
144144 e_max = np .power (10.0 , model_parameters .get ("energy_bins_log10_tev" , []).get ("E_max" )),
145145 )
@@ -165,7 +165,7 @@ def configure_apply(analysis_type):
165165 "--model_prefix" ,
166166 required = True ,
167167 metavar = "MODEL_PREFIX" ,
168- help = ("Path to directory containing XGBoost models (without n_tel / energy bin suffix)." ),
168+ help = ("Path to directory containing XGBoost models (without energy bin suffix)." ),
169169 )
170170 parser .add_argument (
171171 "--model_name" ,
0 commit comments