Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/davidnabergoj/bootplot
Bootplot is a package for black-box uncertainty visualization.
https://github.com/davidnabergoj/bootplot
animation black-box bootstrap bootstrap-sampling images plotting python scientific-visualization statistics uncertainty visualization
Last synced: 3 months ago
JSON representation
Bootplot is a package for black-box uncertainty visualization.
- Host: GitHub
- URL: https://github.com/davidnabergoj/bootplot
- Owner: davidnabergoj
- License: mit
- Created: 2022-04-27T13:04:51.000Z (almost 3 years ago)
- Default Branch: main
- Last Pushed: 2022-08-18T12:20:09.000Z (over 2 years ago)
- Last Synced: 2024-11-02T21:28:41.941Z (3 months ago)
- Topics: animation, black-box, bootstrap, bootstrap-sampling, images, plotting, python, scientific-visualization, statistics, uncertainty, visualization
- Language: Python
- Homepage: https://bootplot.readthedocs.io/en/latest/
- Size: 5.45 MB
- Stars: 2
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
[](https://github.com/davidnabergoj/bootplot/blob/main/LICENSE)
[](https://bootplot.readthedocs.io/en/latest/?badge=latest)
[](https://github.com/davidnabergoj/bootplot/actions/workflows/build.yml)
[](https://github.com/davidnabergoj/bootplot/actions/workflows/tests.yml)
**bootplot** is a package for black-box uncertainty visualization.
By providing a dataset and a plotting function, **bootplot** automatically generates a static image and an animation of your uncertainty.
The method works by resampling the original dataset using bootstrap and plotting each bootstrapped sample.
The plots are then combined into a single image or an animation.
**bootplot** is also especially useful when dealing with small datasets, since it
relies on the bootstrap method which robustly estimates uncertainty using resampling.
**bootplot** supports datasets represented as numpy arrays or pandas dataframes.
Supported image output formats include popular formats such as JPG, PNG, BMP. Supported animation formats include popular formats such as GIF and MP4.
## Installation
**bootplot** requires Python version 3.8 or greater. You can install it using:
```
pip install bootplot
```
Alternatively, you can install **bootplot** using:
```
git clone https://github.com/davidnabergoj/bootplot
cd bootplot
python setup.py install
```
## Example
Suppose we have some data and their corresponding targets. We can model our targets with a regression
line and visualize the uncertainty with the following code:
```python
import numpy as np
from sklearn.linear_model import LinearRegression
from bootplot import bootplot
def plot_regression(data_subset, data_full, ax):
# Plot full dataset
ax.scatter(data_full[:, 0], data_full[:, 1])
# Plot regression line trained on the subset
lr = LinearRegression()
lr.fit(data_subset[:, 0].reshape(-1, 1), data_subset[:, 1])
ax.plot([-10, 10], lr.predict([[-10], [10]]), c='r')
# Show root mean squared error in a text box
rmse = np.sqrt(np.mean(np.square(data_subset[:, 1] - lr.predict(data_subset[:, 0].reshape(-1, 1)))))
bbox_kwargs = dict(facecolor='none', edgecolor='black', pad=10.0)
ax.text(x=0, y=-8, s=f'RMSE: {rmse:.4f}', fontsize=12, ha='center', bbox=bbox_kwargs)
ax.set_xlim(-10, 10)
ax.set_ylim(-10, 10)
if __name__ == '__main__':
np.random.seed(0)
# Dataset to be modeled
dataset = np.random.randn(100, 2)
noise = np.random.randn(len(dataset)) * 2.5
dataset[:, 1] = dataset[:, 0] * 1.5 + 2 + noise
# Create image and animation that show uncertainty
bootplot(
plot_regression,
dataset,
output_image_path='demo_image.png',
output_animation_path='demo_animation.gif',
verbose=True
)
```
This will generate a static image and an animation, as shown below.
The static image on points shows the full scattered dataset in blue and regression lines that correspond to each
bootstrapped sample of the dataset in red.
The spread of regression lines represents uncertainty according to the bootstrap process.
We can also see the uncertainty in root mean squared error (RMSE).
We see that only the first digit of RMSE is significant, since the decimal part is blurred.
The animation on the right displays uncertainty by iterating over a sequence of plots containing regression lines.
See the [examples](examples) folder for more examples, including bar charts, point plots, polynomial regression models, pie charts, text plots and pandas dataframes.
## Documentation
Read the documentation and check out tutorials at https://bootplot.readthedocs.io/en/latest/