| 26 | |
| 27 | @dataclass |
| 28 | class ModelInputData: |
| 29 | T: int |
| 30 | S: int |
| 31 | K: int |
| 32 | tau: float |
| 33 | trend_indicator: int |
| 34 | y: Sequence[float] # length T |
| 35 | t: Sequence[float] # length T |
| 36 | cap: Sequence[float] # length T |
| 37 | t_change: Sequence[float] # length S |
| 38 | s_a: Sequence[int] # length K |
| 39 | s_m: Sequence[int] # length K |
| 40 | X: Sequence[Sequence[float]] # shape (T, K) |
| 41 | sigmas: Sequence[float] # length K |
| 42 | |
| 43 | @dataclass |
| 44 | class ModelParams: |
no outgoing calls
no test coverage detected
searching dependent graphs…