diff --git a/straxen/corrections_services.py b/straxen/corrections_services.py index 71cc26fc9..ef2eda53b 100644 --- a/straxen/corrections_services.py +++ b/straxen/corrections_services.py @@ -16,18 +16,10 @@ corrections_w_file = [ "mlp_model", - "cnn_model", - "gcn_model", "s2_xy_map_mlp", - "s2_xy_map_cnn", - "s2_xy_map_gcn", "s2_xy_map", "s1_xyz_map_mlp", - "s1_xyz_map_cnn", - "s1_xyz_map_gcn", "fdc_map_mlp", - "fdc_map_cnn", - "fdc_map_gcn", "s1_aft_xyz_map", "bayes_model", ] @@ -284,7 +276,7 @@ def get_config_from_cmt(self, run_id, model_type, version="ONLINE"): utilix.mongo_storage.MongoDownloader() :param run_id: run id from runDB - :param model_type: model type and neural network type; model_mlp, or model_gcn or model_cnn + :param model_type: model type and neural network type; model_mlp, or model_cnf :param version: version :param return: NN weights file name diff --git a/straxen/plugins/events/__init__.py b/straxen/plugins/events/__init__.py index 83887d62a..a4e661e9c 100644 --- a/straxen/plugins/events/__init__.py +++ b/straxen/plugins/events/__init__.py @@ -55,12 +55,6 @@ from . import event_s2_positions_mlp from .event_s2_positions_mlp import * -from . import event_s2_positions_cnn -from .event_s2_positions_cnn import * - -from . import event_s2_positions_gcn -from .event_s2_positions_gcn import * - from . import event_s1_positions_cnn from .event_s1_positions_cnn import * diff --git a/straxen/plugins/events/event_s2_positions_cnn.py b/straxen/plugins/events/event_s2_positions_cnn.py deleted file mode 100644 index bd3d4b98e..000000000 --- a/straxen/plugins/events/event_s2_positions_cnn.py +++ /dev/null @@ -1,30 +0,0 @@ -import strax -import straxen -from straxen.plugins.events._event_s2_positions_base import EventS2PositionBase - - -export, __all__ = strax.exporter() - - -@export -class EventS2PositionCNN(EventS2PositionBase): - """CNN for position S2 reconstruction at event level.""" - - algorithm = "cnn" - provides = "event_s2_positions_cnn" - - tf_event_model_cnn = straxen.URLConfig( - default=( - "tf://" - "resource://" - f"cmt://{algorithm}_model" - "?version=ONLINE" - "&run_id=plugin.run_id" - "&fmt=abs_path" - ), - help=( - 'CNN model. Should be opened using the "tf" descriptor. ' - 'Set to "None" to skip computation' - ), - cache=3, - ) diff --git a/straxen/plugins/events/event_s2_positions_gcn.py b/straxen/plugins/events/event_s2_positions_gcn.py deleted file mode 100644 index 0d70bbdbd..000000000 --- a/straxen/plugins/events/event_s2_positions_gcn.py +++ /dev/null @@ -1,30 +0,0 @@ -import strax -import straxen -from straxen.plugins.events._event_s2_positions_base import EventS2PositionBase - - -export, __all__ = strax.exporter() - - -@export -class EventS2PositionGCN(EventS2PositionBase): - """GCN net for position S2 reconstruction at event level.""" - - algorithm = "gcn" - provides = "event_s2_positions_gcn" - - tf_event_model_gcn = straxen.URLConfig( - default=( - "tf://" - "resource://" - f"cmt://{algorithm}_model" - "?version=ONLINE" - "&run_id=plugin.run_id" - "&fmt=abs_path" - ), - help=( - 'MLP model. Should be opened using the "tf" descriptor. ' - 'Set to "None" to skip computation' - ), - cache=3, - ) diff --git a/straxen/plugins/peaks/__init__.py b/straxen/plugins/peaks/__init__.py index e961771df..7bca4104e 100644 --- a/straxen/plugins/peaks/__init__.py +++ b/straxen/plugins/peaks/__init__.py @@ -16,15 +16,9 @@ from . import peak_positions from .peak_positions import * -from . import peak_positions_cnn -from .peak_positions_cnn import * - from . import peak_positions_cnf from .peak_positions_cnf import * -from . import peak_positions_gcn -from .peak_positions_gcn import * - from . import peak_positions_mlp from .peak_positions_mlp import * diff --git a/straxen/plugins/peaks/peak_positions_cnn.py b/straxen/plugins/peaks/peak_positions_cnn.py deleted file mode 100644 index 22c9bf238..000000000 --- a/straxen/plugins/peaks/peak_positions_cnn.py +++ /dev/null @@ -1,31 +0,0 @@ -import strax -import straxen -from straxen.plugins.peaks._peak_positions_base import PeakPositionsBaseNT - - -export, __all__ = strax.exporter() - - -@export -class PeakPositionsCNN(PeakPositionsBaseNT): - """Convolutional Neural Network (CNN) neural net for position reconstruction.""" - - provides = "peak_positions_cnn" - algorithm = "cnn" - __version__ = "0.0.1" - - tf_model_cnn = straxen.URLConfig( - default=( - "tf://" - "resource://" - f"cmt://{algorithm}_model" - "?version=ONLINE" - "&run_id=plugin.run_id" - "&fmt=abs_path" - ), - help=( - 'CNN model. Should be opened using the "tf" descriptor. ' - 'Set to "None" to skip computation' - ), - cache=3, - ) diff --git a/straxen/plugins/peaks/peak_positions_gcn.py b/straxen/plugins/peaks/peak_positions_gcn.py deleted file mode 100644 index df883b51a..000000000 --- a/straxen/plugins/peaks/peak_positions_gcn.py +++ /dev/null @@ -1,31 +0,0 @@ -import strax -import straxen -from straxen.plugins.peaks._peak_positions_base import PeakPositionsBaseNT - - -export, __all__ = strax.exporter() - - -@export -class PeakPositionsGCN(PeakPositionsBaseNT): - """Graph Convolutional Network (GCN) neural net for position reconstruction.""" - - provides = "peak_positions_gcn" - algorithm = "gcn" - __version__ = "0.0.1" - - tf_model_gcn = straxen.URLConfig( - default=( - "tf://" - "resource://" - f"cmt://{algorithm}_model" - "?version=ONLINE" - "&run_id=plugin.run_id" - "&fmt=abs_path" - ), - help=( - 'GCN model. Should be opened using the "tf" descriptor. ' - 'Set to "None" to skip computation' - ), - cache=3, - )