An example project using PipelineX, Kedro, PyTorch, Ignite, and SHAP for image classification
Pipeline visualized by Kedro-viz
$ pip install pipelinex torch torchvision pytorch-ignite kedro shap mlflow kedro-viz
Note: mlflow
and kedro-viz
are optional.
$ git clone https://github.com/Minyus/pipelinex_pytorch.git
$ cd pipelinex_pytorch
$ python main.py
- Python 3.6.8
This project was created from the GitHub template repository at https://github.com/Minyus/pipelinex_template
To use for a new project, fork the template repository and hit Use this template
button next to Clone or download
.