Titanic Project Matchine Learning Model

Titanic ML

Titanic ML is a machine learning project aimed at predicting the survival of passengers on the Titanic using various machine learning models. The project includes data ingestion, transformation, model training, and evaluation.

Table of Contents

• Installation
• Usage
• Features

Installation

To set up the project locally, follow these steps:

1. Clone the repository:

   git clone https://github.com/8BitButter/Titanic---Machine-Learning-from-Disaster

2. Navigate to the project directory:

   cd Titanic_Project

3. Install the required dependencies:

   pip install -r requirements.txt

Usage

To use the project, follow these steps:

1. Ensure you have a MySQL database set up with the necessary data.

2. Create a .env file in the root directory with the following content:

   host=your_host
   user=your_user
   password=your_password
   db=your_database

Features

• Data ingestion from a MySQL database.
• Data transformation using custom pipelines.
• Model training with various algorithms including Random Forest, Decision Tree, Gradient Boosting,Linear Regression, XGBoost, CatBoost, and AdaBoost.
• Model evaluation and selection based on performance metrics.
• Saving the best model for future predictions.