A toolkit for visualizations in materials informatics
Project description
pymatviz
A toolkit for visualizations in materials informatics.
Note: This project is not associated with or endorsed by pymatgen, but aims to complement it by adding additional plotting functionality.
Installation
pip install pymatviz
Usage
Check out the Jupyter notebooks under examples/ to learn how to use pymatviz.
matbench_dielectric_eda.ipynb |
 |
 |
 |
mp_bimodal_e_form.ipynb |
|
|
|
matbench_perovskites_eda.ipynb |
|
|
|
mprester_ptable.ipynb |
|
|
|
When trying to open notebooks in Google Colab, you might encounter errors. Colab currently only supports Python 3.7. pymatviz uses Python 3.8 features like self-documenting f-strings. You may still be able to use pymatviz on Colab by cloning the repo and patching the source code in-place as shown here.
Periodic Table
See pymatviz/ptable.py. Heat maps of the periodic table can be plotted both with matplotlib and plotly. plotly supports displaying additional data on hover or full interactivity through Dash.
ptable_heatmap(compositions, log=True) |
ptable_heatmap_ratio(comps_a, comps_b) |
|---|---|
 |
 |
ptable_heatmap_plotly(atomic_masses) |
ptable_heatmap_plotly(compositions) |
 |
 |
Dash app using ptable_heatmap_plotly()
See examples/mprester_ptable.ipynb.
Sunburst
See pymatviz/sunburst.py.
spacegroup_sunburst([65, 134, 225, ...]) |
spacegroup_sunburst(["C2/m", "P-43m", "Fm-3m", ...]) |
|---|---|
 |
 |
Sankey
See pymatviz/sankey.py.
sankey_from_2_df_cols(df_perovskites) |
sankey_from_2_df_cols(df_rand_ints) |
|---|---|
 |
 |
Structure
See pymatviz/structure_viz.py. Currently structure plotting is only supported with matplotlib in 2d. 3d interactive plots (probably with plotly) are on the road map.
plot_structure_2d(mp_19017) |
plot_structure_2d(mp_12712) |
|---|---|
 |
 |
Histograms
spacegroup_hist([65, 134, 225, ...]) |
spacegroup_hist(["C2/m", "P-43m", "Fm-3m", ...]) |
|---|---|
 |
 |
residual_hist(y_true, y_pred) |
hist_elemental_prevalence(compositions, log=True, bar_values='count') |
 |
 |
Parity Plots
See pymatviz/parity.py.
Uncertainty Calibration & Estimation
qq_gaussian(y_true, y_pred, y_std) |
qq_gaussian(y_true, y_pred, y_std: dict) |
|---|---|
 |
 |
error_decay_with_uncert(y_true, y_pred, y_std) |
error_decay_with_uncert(y_true, y_pred, y_std: dict) |
 |
 |
Cumulative Error and Residual
cumulative_error(preds, targets) |
cumulative_residual(preds, targets) |
|---|---|
 |
 |
Classification Metrics
roc_curve(targets, proba_pos) |
precision_recall_curve(targets, proba_pos) |
|---|---|
 |
 |
Correlation
marchenko_pastur(corr_mat, gamma=ncols/nrows) |
marchenko_pastur(corr_mat_significant_eval, gamma=ncols/nrows) |
|---|---|
 |
 |
Glossary
- Residual
y_res = y_true - y_pred: The difference between ground truth target and model prediction. - Error
y_err = abs(y_true - y_pred): Absolute error between target and model prediction. - Uncertainty
y_std: The model's estimate for its error, i.e. how much the model thinks its prediction can be trusted. (stdfor standard deviation.)
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
File details
Details for the file pymatviz-0.5.0.tar.gz.
File metadata
- Download URL: pymatviz-0.5.0.tar.gz
- Upload date:
- Size: 43.5 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.1 CPython/3.8.13
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | c65e057fc8c1f33ceaece1787e96df18b1170460acdeb37a2a9f5dcdf45bd543 |
|
| MD5 | e9faa1b49a60295aafd747497d9fb133 |
|
| BLAKE2b-256 | 1b3d64597fefd005205ead6f99c27cb70ceb38485cbe6c0e5bf784417e857069 |
