No project description provided
Project description
scikit-stan
Scikit-Stan is a package of Stan models wrapped in a Scikit-Learn style interface.
This package is currently under active development and should be treated as beta software.
Documentation is available at https://brianward.dev/scikit-stan/
Installation
Pre-compiled wheels for the package are available for MacOS, Windows, and Linux systems via pip install scikit_stan
.
Source installation requires a working installation of CmdStan.
Basic usage
from scikit_stan import GLM
m = GLM(family='gamma') # Gamma family distribution with canonical inverse link
m.fit(X, y) # runs HMC-NUTS
predictions = m.predict(X) # generates new predictions from fitted model
score = m.score(X, y) # computes the R2 score of the fitted model on the data X and observations y
Attribution
This package is licensed under the BSD 3-clause license.
It is inspired by existing packages in the Stan ecosystem like rstanarm.
This package was initially developed at the Simons Foundation by Alexey Izmailov during a summer 2022 internship under the mentorship of Brian Ward.
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 Distributions
Hashes for scikit_stan-0.1.0-py3-none-win_amd64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | f36f1457bb8d063b016b77c67edcd73c4587167b5cf0c799f5c3cf8c6c19f84f |
|
MD5 | b0189b4e1c645c554fc319c4c04338ef |
|
BLAKE2b-256 | e157354d4ee91a99d14cf586ebc9663e274a57edc1f2d59117e6ce0ccf776dad |
Hashes for scikit_stan-0.1.0-py3-none-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 8e98b6cbcfd73e02190eacebea7a4434bdb5fdcd004546d94bc96c753a0fef6a |
|
MD5 | c574818468e45c1f366d979705b565c2 |
|
BLAKE2b-256 | cbb61f49eb92d5cd18ff7515212301742ac019357723ce8050c30c1b28f4db8a |
Hashes for scikit_stan-0.1.0-py3-none-macosx_10_9_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | fa8c4b8051fe808d650f20adf04108311fcb2adcd38873082a9cccf41cd519c3 |
|
MD5 | 8228359e7110d3865f22d7ef4e98d031 |
|
BLAKE2b-256 | 07b165fb4205015d7de7827c44ac0492df7d395181d41477df5021f61a6f7721 |