import strax
import straxen
from straxen.plugins.events._event_s2_positions_base import EventS2PositionBase
export, __all__ = strax.exporter()
[docs]@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,
)