Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Forgis-Labs/HEPA
/ types & classes
Types & classes
10 in github.com/Forgis-Labs/HEPA
⨍
Functions
97
◇
Types & classes
10
↓ 8 callers
Class
HEPA
Horizon-conditioned Event Predictive Architecture. Components: encoder causal Transformer over context, returns h_t targe
hepa/model/hepa.py:25
↓ 2 callers
Class
PatchEmbedding
Group P timesteps across all C channels into one token.
hepa/model/encoder.py:71
↓ 2 callers
Class
PretrainDataset
Random-cut (context, target, Delta_t) sampler over a corpus of sequences. For each sequence we sample n_cuts pairs (t, Delta_t) with Delta_t draw
hepa/training/pretrain.py:34
↓ 2 callers
Class
RevIN
Reversible per-context, per-channel z-score normalization (Kim et al. 2022).
hepa/model/encoder.py:38
↓ 2 callers
Class
_TransformerBlock
Pre-norm Transformer encoder block.
hepa/model/encoder.py:91
↓ 1 callers
Class
CausalEncoder
Causal Transformer encoder over patched, RevIN-normalized inputs. Returns the last valid token h_t as the context summary.
hepa/model/encoder.py:125
↓ 1 callers
Class
EventDataset
Sliding-window dataset emitting (context, time_to_event, t_index).
hepa/training/finetune.py:26
↓ 1 callers
Class
EventHead
LayerNorm + linear to a single scalar logit per horizon.
hepa/model/event_head.py:9
↓ 1 callers
Class
HorizonPredictor
MLP that maps (context embedding, horizon) to a predicted future embedding.
hepa/model/predictor.py:9
↓ 1 callers
Class
TargetEncoder
Bidirectional Transformer over the target interval, then attention pooling.
hepa/model/target_encoder.py:29