Skip to content

Basic example to use PipelineX, PyTorch, Ignite, Shap, Kedro, MLflow

License

Notifications You must be signed in to change notification settings

Minyus/pipelinex_pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

29 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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.

About

Basic example to use PipelineX, PyTorch, Ignite, Shap, Kedro, MLflow

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages