A time-series forecasting model using UR2CUTE: Using Repetitively 2 CNN for Unsteady Timeseries Estimation
Project description
UR2CUTE
Using Repetitively 2 CNNs for Unsteady Timeseries Estimation. UR2CUTE is a dual-stage, PyTorch powered model dedicated to intermittent demand forecasting. A classifier estimates demand occurrence while a regressor predicts magnitude, allowing the library to focus on the sparse structure typical of slow-moving inventory.
What's New in 2.0
Version 2.0.0 keeps the two-CNN hurdle design but overhauls how the networks see the data. This is a breaking release.
- Multichannel temporal windows replace the old flat lag vector. Each CNN now receives
(channels, window)where the window axis is genuine time: target history, optional covariates, and engineered intermittency channels (time-since-last-demand, causal rolling mean). - Dilated causal convolutions (TCN-style) so the convolutions slide over consecutive time steps and cover long inter-demand gaps cheaply.
- Better training objectives:
BCEWithLogitsLosswith per-steppos_weightfor the classifier (heavy zero-class imbalance) and a Huber (SmoothL1) loss for the regressor (robust to spikes). - Smarter scaling: per-channel min-max plus an automatic
log1ptransform on non-negative targets.
Migration notes:
- Models saved with 1.x cannot be loaded in 2.0 — retrain and re-save.
- The default
n_steps_lagchanged from3to12(it is now a lookback window, not a count of lag columns). Set it explicitly to pin behaviour.
What's New in 2.2
Better out-of-the-box defaults, informed by benchmarking on real intermittent data:
thresholdnow defaults to"balanced"(was0.5). It tunes the occurrence cutoff on the validation split to drive forecast bias toward zero — the previous fixed/"auto"cutoffs systematically under-forecast. Adds no training cost.verbosenow defaults toFalse(wasTrue), which suits fitting many series in a loop. Passverbose=Trueto restore per-epoch logging.
Both are behaviour changes for code relying on the old defaults; set the parameters explicitly to pin previous behaviour.
Overview
Intermittent demand is dominated by long zero stretches punctuated by irregular spikes. Traditional statistical models struggle to capture both the timing and the size of those bursts. UR2CUTE tackles the problem with a hurdle-style architecture:
- A dilated-causal CNN classifier predicts the probability of non-zero demand for each step in the forecast horizon.
- A dilated-causal CNN regressor estimates the corresponding quantities.
- Final forecasts combine the two outputs through an adaptive threshold so the regressor only contributes when demand is likely.
Both CNNs consume the same multichannel temporal window — target history, optional covariates, and engineered intermittency channels — so the convolutions slide over a genuine time axis and capture long inter-demand gaps cheaply.
The estimator follows the scikit-learn API, includes thorough input validation, and automatically selects CPU or GPU devices.
Features
- Pure PyTorch implementation with GPU support when available.
- Direct multi-step forecasting: predicts the entire horizon in a single forward pass.
- Multichannel temporal windows: target history, optional external covariates, and engineered intermittency channels (time-since-last-demand, causal rolling mean).
- Dilated causal convolutions (TCN-style) so each CNN slides over a genuine time axis and cheaply covers long inter-demand gaps.
- Customizable hyperparameters (epochs, batch size, independent learning rates, dropout).
- Occurrence threshold modes: a fixed value,
"auto"(fraction of zeros), or"balanced"(tuned on validation to drive the forecast bias toward zero). - Early stopping that restores the best weights, kept in memory during training.
- Reproducible results through explicit random seed management.
- Model persistence through
save_modelandload_model. - Complete type hints and packaged type information (
py.typed).
Dependencies
- Python 3.7 or newer
- PyTorch 1.7+
- NumPy 1.20+ (for
sliding_window_view) - pandas
- scikit-learn
Installation
From PyPI
pip install UR2CUTE
From Source
git clone https://github.com/FH-Prevail/UR2CUTE_torch.git
cd UR2CUTE_torch
pip install -e .
# Optional extras
pip install -e ".[dev]"
pip install -e ".[test]"
pip install -e ".[docs]"
Verify Installation
from UR2CUTE import UR2CUTE
print(UR2CUTE.__module__)
Quick Start
import pandas as pd
import torch
from UR2CUTE import UR2CUTE
data = pd.DataFrame(
{
"date": pd.date_range("2023-01-01", periods=50, freq="W"),
"target": [0, 5, 0, 0, 12, 0, 0, 0, 7, 0] * 5,
"promo": [0, 1, 0, 0, 1, 0, 0, 1, 0, 0] * 5,
"price": [10.0, 9.5, 9.5, 9.5, 10.0, 10.0, 10.0, 9.8, 9.8, 9.8] * 5,
}
)
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
print(f"Using device: {device}")
model = UR2CUTE(
n_steps_lag=12,
forecast_horizon=4,
external_features=["promo", "price"],
threshold="balanced",
)
model.fit(data, target_col="target")
print(model.predict(data))
Parameters
| Parameter | Description | Default |
|---|---|---|
n_steps_lag |
Length of the lookback window (past time steps each CNN sees). | 12 |
forecast_horizon |
Number of future periods predicted per call. | 8 |
external_features |
Optional list of column names used as exogenous inputs. | None |
epochs |
Training epochs for both CNN models. | 100 |
batch_size |
Training batch size. | 32 |
threshold |
Occurrence cutoff: a float, "auto" (fraction of zeros), or "balanced" (tuned on validation to minimize forecast bias). |
"balanced" |
patience |
Early stopping patience (epochs). | 10 |
random_seed |
Global random seed applied to NumPy, Python, and PyTorch. | 42 |
classification_lr |
Learning rate for the classifier. | 0.0021 |
regression_lr |
Learning rate for the regressor. | 0.0021 |
dropout_classification |
Dropout applied inside the classifier. | 0.4 |
dropout_regression |
Dropout applied inside the regressor. | 0.2 |
regressor_nonzero_only |
When True, the regressor is trained only on sequences where the forecast horizon contains at least one non-zero value. Set to False to train the regressor on all sequences. |
True |
verbose |
Enables progress output and early-stopping logs. | False |
Usage Patterns
Auto Threshold
model = UR2CUTE(threshold="auto")
model.fit(df, "target")
print(model.threshold_)
External Features
covariates = ["promotion", "price", "weekday"]
model = UR2CUTE(external_features=covariates)
model.fit(df, "target")
Progress Logging
Training is silent by default; pass verbose=True for per-epoch loss and
early-stopping logs.
model = UR2CUTE(verbose=True)
model.fit(df, "target")
Model Persistence
trained = UR2CUTE().fit(train_df, "target")
trained.save_model("production_model.pkl")
loaded = UR2CUTE.load_model("production_model.pkl")
preds = loaded.predict(new_df)
How It Works
- Preprocessing – validates the input frame, builds the multichannel feature matrix (target + covariates + engineered intermittency channels), slides it into multi-step temporal windows, and splits chronologically into train and validation partitions.
- Scaling – fits per-channel min-max scaling on the training windows only, and applies a
log1ptransform to non-negative targets to tame spikes before scaling, preventing validation leakage. - Classification Stage – trains a dilated-causal CNN with
BCEWithLogitsLossand per-steppos_weightto estimate the probability of demand for each future horizon step under heavy zero-class imbalance. - Regression Stage – trains a dilated-causal CNN regressor with a Huber (SmoothL1) loss, robust to demand spikes. By default (
regressor_nonzero_only=True) only sequences where the horizon contains at least one non-zero value are used, keeping the regressor focused on demand magnitude; if no such sequences exist it falls back to the full dataset. Setregressor_nonzero_only=Falseto train on all sequences instead. - Inference – transforms the latest observed window, runs both networks, rescales quantities, and zeros out forecasts whose occurrence probability falls below the stored threshold.
Performance
Internal benchmarks show UR2CUTE outperforming Croston, AutoARIMA, Prophet, gradient boosted trees, and random forests on sparse demand series, especially in MAE% and RMSE%. Improvements stem from the dedicated occurrence model, the multichannel temporal window with engineered intermittency signals, and the dilated causal filters tuned to each dataset.
Citation
@article{mirshahi2024intermittent,
title={Intermittent Time Series Demand Forecasting Using Dual Convolutional Neural Networks},
author={Mirshahi, Sina and Brandtner, Patrick and Kominkova Oplatkova, Zuzana},
journal={MENDEL -- Soft Computing Journal},
volume={30},
number={1},
year={2024},
publisher={MENDEL Journal}
}
License
UR2CUTE is released under the MIT License. See LICENSE for the full text.
Contributors
- Sina Mirshahi
- Patrick Brandtner
- Zuzana Kominkova Oplatkova
- Taha Falatouri
- Mehran Naseri
- Farzaneh Darbanian
Acknowledgments
This work was carried out at:
- Department of Informatics and Artificial Intelligence, Tomas Bata University
- Department for Logistics, University of Applied Sciences Upper Austria, Steyr
- Josef Ressel-Centre for Predictive Value Network Intelligence, Steyr
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file ur2cute-2.2.0.tar.gz.
File metadata
- Download URL: ur2cute-2.2.0.tar.gz
- Upload date:
- Size: 22.1 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.14.5
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
132e7c21f91e3dfc6cf5ae5232683b418a1b005dc4ec014672d23c6856179517
|
|
| MD5 |
077c1b02fece70ee48fef6eb7832d023
|
|
| BLAKE2b-256 |
ac672a6f18427599f70a2d09578b3366b578eadf56d964ca14fa245e05e514f3
|
File details
Details for the file ur2cute-2.2.0-py3-none-any.whl.
File metadata
- Download URL: ur2cute-2.2.0-py3-none-any.whl
- Upload date:
- Size: 17.1 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.14.5
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
feba4ddf853b42f3d827f73a57fd5e197dec56611adf587d0f3788e7b127507f
|
|
| MD5 |
94ee5449bca2a761e913ac239b5523c7
|
|
| BLAKE2b-256 |
3f16df65d09c9c5b05dd5c0a3f8cbabb6fd29ffdf1f49da76966154b24bca0bd
|