Customize components

pruna is designed to be customizable. You can add your own algorithms, datasets, and metrics to the package.

Customize an algorithm

Steps to integrate a new compression algorithm, making it available in the SmashConfig.

./customize_algorithm.rst

Customize a dataset

Steps to integrate a new dataset, making it available in the SmashConfig.

./customize_dataset.rst

Customize a metric

Steps to integrate a new metric, making it available in the SmashConfig.

./customize_metric.rst

Tip

You can also customize the package by adding your own algorithms, datasets, and metrics. Take a look at the contributing guide to learn more. If anything is unclear or you want to discuss your contribution before opening a PR, please reach out on Discord anytime!