Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/stan-dev/stanio

Preparing inputs to and reading outputs from Stan
https://github.com/stan-dev/stanio

csv io json python stan

Last synced: 2 months ago
JSON representation

Preparing inputs to and reading outputs from Stan

Awesome Lists containing this project

README

        

# StanIO

[![codecov](https://codecov.io/gh/stan-dev/stanio/graph/badge.svg?token=P93MLO21FK)](https://codecov.io/gh/stan-dev/stanio) [![Tests](https://github.com/stan-dev/stanio/actions/workflows/test.yaml/badge.svg)](https://github.com/stan-dev/stanio/actions/workflows/test.yaml)

A set of Python functions for data-wrangling in the formats used by
the [Stan](https://mc-stan.org) probabalistic programming language.

It is primarily developed for use in [cmdstanpy](https://github.com/stan-dev/cmdstanpy).

## Features

- Writing Python dictionaries to Stan-compatible JSON (with optional support for using `ujson` for faster serialization)
- Basic reading of StanCSV files into numpy arrays
- Parameter extraction from numpy arrays based on StanCSV headers
- e.g., if you have a `matrix[2,3] x` in a Stan program, extracting `x` will give you a `(num_draws, 2, 3)` numpy array