Skip to main content

PSANN: Parameterized Sine-Activated Neural Networks (sklearn-style, PyTorch backend)

Project description

PSANN - Parameterized Sine-Activated Neural Networks

Sklearn-style estimators built on PyTorch that use learnable sine activations, optional persistent state, and shared helpers for episodic (HISSO) training.

Quick links:

  • API reference: docs/API.md
  • Technical design notes: TECHNICAL_DETAILS.md
  • Scenario walkthroughs: docs/examples/README.md

Installation

python -m venv .venv
.\.venv\Scripts\Activate.ps1   # Windows PowerShell
# source .venv/bin/activate     # macOS/Linux
pip install --upgrade pip
pip install -e .                # editable install from source

Extras defined in pyproject.toml:

  • psann[sklearn]: adds scikit-learn for real BaseEstimator mixins and metrics
  • psann[viz]: plotting helpers used in notebooks/examples
  • psann[dev]: pytest, ruff, black

Need pre-pinned builds (e.g. on Windows or air-gapped envs)? Use the compatibility constraints:

pip install -e . -c requirements-compat.txt

pyproject.toml is the authoritative dependency list. requirements-compat.txt mirrors the newest widely available wheels for NumPy, SciPy, and scikit-learn when you need lockstep installs.

NumPy 2.0 status

  • Core PSANN modules run cleanly on NumPy 2.0; no removed aliases (e.g. np.bool, np.int) are referenced.
  • The base install now depends on numpy>=1.23 with no upper cap. Use requirements-compat.txt when you need the older NumPy 1.26 wheels for legacy environments.
  • The optional psann[sklearn] extra still uses scikit-learn<1.5; those wheels currently target NumPy 1.x. Install scikit-learn>=1.5 when available or stick to the compatibility constraints if you rely on that extra.

Quick Start

Supervised regression

import numpy as np
from psann import PSANNRegressor

rs = np.random.RandomState(42)
X = np.linspace(-4, 4, 1000, dtype=np.float32).reshape(-1, 1)
y = 0.8 * np.exp(-0.25 * np.abs(X)) * np.sin(3.5 * X)

model = PSANNRegressor(
    hidden_layers=2,
    hidden_units=64,
    epochs=200,
    lr=1e-3,
    early_stopping=True,
    patience=20,
    random_state=42,
)
model.fit(X, y, verbose=1)
print("R^2:", model.score(X, y))

Supervising extras outputs

Append extra columns to y or pass them separately. Extra heads are scheduled automatically when extras>0.

extras = np.stack([np.cos(X[:, 0]), np.sin(X[:, 0])], axis=1).astype(np.float32)
y_with_extras = np.concatenate([y[:, None], extras], axis=1)

model = PSANNRegressor(hidden_layers=2, hidden_units=64, extras=2)
model.fit(X, y_with_extras, verbose=1)

For streaming/time-series, LSM preprocessors, segmentation heads, and HISSO workflows, head to docs/examples/README.md.

Feature Highlights

  • Learnable sine activations (SineParam) with amplitude, frequency, and decay bounds
  • Shared _fit helper powering PSANN, residual PSANN, and language-model estimators
  • Optional predictive extras with automatic target detection and rollout utilities
  • Stateful controllers for streaming inference with warm-start and reset policies
  • Convolutional variants that preserve spatial structure and support per-element outputs
  • HISSO episodic training with reward hooks, supervised warm starts, and extras scheduling

Contributing

  • Run pip install -e .[dev] and then pytest / ruff check .
  • Format code with black and keep docstrings ASCII unless a file already uses Unicode
  • See open TODO items in TODO.txt

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

psann-0.9.15.tar.gz (146.7 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

psann-0.9.15-py3-none-any.whl (69.5 kB view details)

Uploaded Python 3

File details

Details for the file psann-0.9.15.tar.gz.

File metadata

  • Download URL: psann-0.9.15.tar.gz
  • Upload date:
  • Size: 146.7 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.11.9

File hashes

Hashes for psann-0.9.15.tar.gz
Algorithm Hash digest
SHA256 8c4973e010c264969156fdd8dddedb129c398b69e6b0c483c0442d9c98d0cfe6
MD5 d4fea12805ea0d72f324d980e63bd279
BLAKE2b-256 51b612630ce28be221b76edb30e4f92cc3f5143aba3a0593e9b9ace0c2863cda

See more details on using hashes here.

File details

Details for the file psann-0.9.15-py3-none-any.whl.

File metadata

  • Download URL: psann-0.9.15-py3-none-any.whl
  • Upload date:
  • Size: 69.5 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.11.9

File hashes

Hashes for psann-0.9.15-py3-none-any.whl
Algorithm Hash digest
SHA256 4f63f0b80fbff6ae803449bb397a4a81ab5fa3d86fbcbaa79921966417a38f71
MD5 64d9801a61094808113a837f702f867f
BLAKE2b-256 8837b2ca4fe253769c9fef3059ad1e7fe889283c2ecfe4e2758e77b917c9decb

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page