Welcome to WaveletML’s documentation!

https://img.shields.io/badge/release-0.2.0-yellow.svg https://badge.fury.io/py/waveletml.svg https://img.shields.io/pypi/pyversions/waveletml.svg https://img.shields.io/pypi/dm/waveletml.svg https://github.com/thieu1995/WaveletML/actions/workflows/publish-package.yml/badge.svg https://pepy.tech/badge/waveletml https://readthedocs.org/projects/waveletml/badge/?version=latest https://img.shields.io/badge/Chat-on%20Telegram-blue https://img.shields.io/badge/DOI-10.6084%2Fm9.figshare.28802531-blue https://img.shields.io/badge/License-GPLv3-blue.svg

WaveletML is an open-source Python framework designed for building, training, and evaluating Wavelet Neural Networks (WNNs) tailored for supervised learning tasks such as regression and classification. Leveraging the power of PyTorch and the modularity of scikit-learn, WaveletML provides a unified, extensible, and scalable platform for researchers and practitioners to explore wavelet-based neural architectures.

  • Free software: GNU General Public License (GPL) V3 license

  • Provided Estimators: GdWnnClassifier, GdWnnRegressor, MhaWnnClassifier, MhaWnnRegressor

  • Supported Wavelet Functions: Morlet, Mexican Hat, Haar, …

  • Supported Wavelet Layers: - Weighed Linear Layer - Product Layer - Summation Layer

  • Supported performance metrics: >= 67 (47 regressions and 20 classifications)

  • Documentation: https://waveletml.readthedocs.io

  • Python versions: >= 3.8.x

  • Dependencies: numpy, scipy, scikit-learn, pandas, permetrics, mealpy, torch

Indices and tables