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