liamsbhoo commited on
Commit
f8a4cfd
·
verified ·
1 Parent(s): 5363264

Add model card

Browse files
Files changed (1) hide show
  1. README.md +49 -0
README.md ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - time-series
5
+ - time-series-classification
6
+ - in-context-learning
7
+ library_name: timee
8
+ ---
9
+
10
+ # TIMEE — Multivariate (beta)
11
+
12
+ Multivariate checkpoint for [TIMEE](https://github.com/automl/timee), fine-tuned for
13
+ **attention-based variate pooling**. Each variate is encoded through the shared univariate
14
+ encoder, and the per-variate representations are fused by a learned attention pool
15
+ (`variate_attn_pool`) before the in-context phase — modeling channels jointly rather than
16
+ independently.
17
+
18
+ > **Beta:** TIMEE is trained and evaluated as a univariate classifier. Multivariate is not
19
+ > its focus (yet) — this checkpoint is provided so people can use and evaluate it. The main
20
+ > univariate model lives at [`liamsbhoo/timee`](https://huggingface.co/liamsbhoo/timee).
21
+
22
+ ## Usage
23
+
24
+ ```python
25
+ from timee import TimeeMultivariateClassifier
26
+
27
+ clf = TimeeMultivariateClassifier.from_pretrained("liamsbhoo/timee-multivariate")
28
+
29
+ # X: (n_samples, n_channels, seq_len) float32
30
+ predictions, probabilities = clf.predict(X_train, y_train, X_test)
31
+ ```
32
+
33
+ For zero-shot multivariate classification without this checkpoint, `TimeeClassifier`
34
+ handles `n_channels > 1` by classifying each channel independently and averaging the
35
+ per-channel class probabilities.
36
+
37
+ ## Citation
38
+
39
+ ```bibtex
40
+ @misc{küken2026timeeendtoendtimeseries,
41
+ title={TimEE: End-to-end Time Series Classification via In-Context Learning},
42
+ author={Jaris Küken and Shi Bin Hoo and Martin Mráz and Frank Hutter and Lennart Purucker},
43
+ year={2026},
44
+ eprint={2607.07500},
45
+ archivePrefix={arXiv},
46
+ primaryClass={cs.LG},
47
+ url={https://arxiv.org/abs/2607.07500},
48
+ }
49
+ ```