Skip to content
Snippets Groups Projects
Commit 7a38ac2d authored by Danilo Ferreira de Lima's avatar Danilo Ferreira de Lima
Browse files

Added method to get channels used in training.

parent df5e42c9
No related branches found
No related tags found
No related merge requests found
......@@ -532,6 +532,10 @@ class Model(TransformerMixin, BaseEstimator):
# size of the test subset
self.validation_size = validation_size
def get_channels(self) -> List[str]:
"""Get channels used in training."""
return self.x_model.named_steps["select"].channels
def debug_peak_finding(self, low_res_data: Dict[str, np.ndarray], filename: str):
"""
Produce image to understand if the peak finding step worked well.
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment