Skip to content

Latest commit

 

History

History
39 lines (25 loc) · 1007 Bytes

README.md

File metadata and controls

39 lines (25 loc) · 1007 Bytes
Raw

PipelineX PyTorch

An example project using PipelineX, Kedro, PyTorch, Ignite, and SHAP for image classification

Pipeline visualized by Kedro-viz

1. Install dependencies

$ pip install pipelinex torch torchvision pytorch-ignite kedro shap mlflow kedro-viz

Note: mlflow and kedro-viz are optional.

2. Clone this repository and run main.py

$ git clone https://github.com/Minyus/pipelinex_pytorch.git
$ cd pipelinex_pytorch
$ python main.py

Tested environment

  • Python 3.6.8

Simplified Kedro project template

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.