https://github.com/rafat3000/deep-learning-with-mlflow

https://github.com/rafat3000/deep-learning-with-mlflow

classification deep-learning github mlflow python

Last synced: about 1 month ago
JSON representation

• Host: GitHub
• URL: https://github.com/rafat3000/deep-learning-with-mlflow
• Owner: Rafat3000
• Created: 2024-11-21T10:19:52.000Z (11 months ago)
• Default Branch: main
• Last Pushed: 2024-11-21T10:25:45.000Z (11 months ago)
• Last Synced: 2025-04-09T16:53:06.015Z (6 months ago)
• Topics: classification, deep-learning, github, mlflow, python
• Language: Jupyter Notebook
• Homepage:
• Size: 44.9 KB
• Stars: 4
• Watchers: 1
• Forks: 1
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

          
# Deep-Learning-with-MLFlow

# Gender Classification with Neural Network and MLflow

This project demonstrates how to build a deep learning model using TensorFlow for gender classification and integrate it with MLflow for experiment tracking. The model is trained on a pre-processed dataset, and the experiments are logged to a local MLflow server.

---

## Features

- **Data Preprocessing**: Handles categorical data using `LabelEncoder` and scales features using `StandardScaler`.

- **Neural Network Architecture**: A multi-layer dense neural network built with TensorFlow/Keras.

- **Experiment Tracking**: Logs parameters, metrics, and model artifacts to MLflow.

---

## Requirements

To run this project, you need the following:

- Python 3.7+

- TensorFlow

- MLflow

- pandas

- scikit-learn