liamsbhoo commited on
Commit
f870da3
·
verified ·
1 Parent(s): 46ce385

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +66 -0
README.md CHANGED
@@ -1,3 +1,69 @@
1
  ---
 
 
 
 
 
 
 
2
  license: apache-2.0
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
+ language:
3
+ - en
4
+ tags:
5
+ - time-series
6
+ - classification
7
+ - in-context-learning
8
+ - transformer
9
  license: apache-2.0
10
  ---
11
+
12
+ # TIMEE: Time Series Classification via In-Context Learning
13
+
14
+ TIMEE is a pretrained transformer for time series classification.
15
+ It classifies test series in a **single forward pass** given labeled training examples — no per-dataset training or fine-tuning required.
16
+
17
+ **Paper:** *TIMEE: End-to-end Time Series Classification via In-Context Learning* (under review)
18
+
19
+ ## Usage
20
+
21
+ ```bash
22
+ pip install timee
23
+ ```
24
+
25
+ ```python
26
+ from timee import TimeeClassifier
27
+ import numpy as np
28
+
29
+ # Downloads weights automatically on first use
30
+ clf = TimeeClassifier.from_pretrained()
31
+
32
+ # X: (n_samples, n_channels, seq_len) float32
33
+ X_train = np.random.randn(20, 1, 256).astype(np.float32)
34
+ y_train = np.array([0, 1] * 10)
35
+ X_test = np.random.randn(5, 1, 256).astype(np.float32)
36
+
37
+ predictions, probabilities = clf.predict(X_train, y_train, X_test)
38
+ ```
39
+
40
+ Labels can be any type (`int`, `str`, etc.).
41
+ Datasets with more than 10 classes are handled automatically via one-vs-rest.
42
+
43
+ ## UCR Benchmark Results
44
+
45
+ | Dataset | Classes | Accuracy | ROC AUC |
46
+ |--------------------|---------|----------|---------|
47
+ | ArrowHead | 3 | 76.6 % | 0.964 |
48
+ | ECG5000 | 5 | 95.0 % | 0.952 |
49
+ | GunPoint | 2 | 98.7 % | 0.997 |
50
+ | ItalyPowerDemand | 2 | 96.0 % | 0.993 |
51
+ | TwoPatterns | 4 | 99.8 % | 1.000 |
52
+
53
+ Results use the default 4-member ensemble (interpolate × {256, 512} × {raw, first-difference}).
54
+
55
+ ## Model Details
56
+
57
+ - **Parameters:** 4,557,322
58
+ - **Input:** univariate or multivariate time series, any length
59
+ - **Output:** class probabilities over up to 10 classes (OvR for more)
60
+ - **Inference:** single forward pass; no dataset-specific adaptation
61
+
62
+ ## Citation
63
+
64
+ ```bibtex
65
+ @inproceedings{timee2026,
66
+ title = {TIMEE: End-to-end Time Series Classification via In-Context Learning},
67
+ year = {2026},
68
+ }
69
+ ```