# training_kit

Wrappers for scikit-learn and PyTorch models with OpenVINO optimization.

## About

This module provides easy-to-use wrappers for training, evaluating, and exporting classical (scikit-learn) and deep learning (PyTorch) models optimized for OpenVINO, targeting local AI PCs and edge deployment.

## Installation

```bash
pip install training_kit
```

## Usage

For detailed usage instructions and examples, please refer to the README files inside the `src/sklearn` and `src/pytorch` folders.

---

For questions, suggestions, or contributions, feel free to open an issue or pull

## 🎓 Credits & License

Developed as part of a GSoC

### Authors

- Leonardo Heim 
- Shivam Basia 
