Creating a Dockerized stack environment using MLflow, mysql and Minio to manage the lifecycle of TensorFlow models.
-
Updated
Mar 15, 2024 - Python
Creating a Dockerized stack environment using MLflow, mysql and Minio to manage the lifecycle of TensorFlow models.
We built a text classification model to predict movie review sentiment using the IMDB dataset. Pre-trained GNews Swivel embeddings from TensorFlow Hub were used for better performance. A simple neural network was trained for 5 epochs, achieving high validation accuracy.
Add a description, image, and links to the swivel topic page so that developers can more easily learn about it.
To associate your repository with the swivel topic, visit your repo's landing page and select "manage topics."