← all repositories

maxjcohen/transformer

A PyTorch implementation of Transformer neural networks adapted for time series forecasting and building consumption prediction.

906 stars Jupyter Notebook Domain AppsML Frameworks
transformer
Velocity · 7d
+0.4
★ / day
Trend
steady
star history

This repository implements the Transformer architecture (originally from NLP) applied to time series prediction. It adapts the model by replacing embeddings with linear layers, using modified positional encodings that better match day/night patterns, and applying windowed attention to focus on short-term patterns. The use case is modeling a numerical simulator for building energy consumption prediction.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.