finetuner 0.0.3.dev13

Finetuner allows one to finetune any deep neural network for better embedding on search tasks

Finetuning any deep neural network for better embedding on neural search tasks

Finetuner allows one to tune the weights of any deep neural network for better embedding on search tasks. It accompanies Jina to deliver the last mile of performance-tuning for neural search applications.

🔱 Powerful yet intuitive: all you need is finetuner.fit(), a one-liner that unlocks rich features such as siamese/triplet architecture, interactive labeling, weight freezing, dimensionality reduction.

⚛️ Framework-agnostic: support Pytorch, Keras and PaddlePaddle as deep learning backends with identical user experience.

🧈 Jina integration: buttery smooth integration with Jina, reducing the cost of context-switch between experimenting and production.

Install

Make sure you have Python 3.7+ and one of Pytorch, Keras or PaddlePaddle installed on Linux/MacOS.

pip install finetuner

Documentation