NYC Data Science Academy| Blog
Bootcamps
Lifetime Job Support Available Financing Available
Bootcamps
Data Science with Machine Learning Flagship 🏆 Data Analytics Bootcamp Artificial Intelligence Bootcamp New Release 🎉
Free Lesson
Intro to Data Science New Release 🎉
Find Inspiration
Find Alumni with Similar Background
Job Outlook
Occupational Outlook Graduate Outcomes Must See 🔥
Alumni
Success Stories Testimonials Alumni Directory Alumni Exclusive Study Program
Courses
View Bundled Courses
Financing Available
Bootcamp Prep Popular 🔥 Data Science Mastery Data Science Launchpad with Python View AI Courses Generative AI for Everyone New 🎉 Generative AI for Finance New 🎉 Generative AI for Marketing New 🎉
Bundle Up
Learn More and Save More
Combination of data science courses.
View Data Science Courses
Beginner
Introductory Python
Intermediate
Data Science Python: Data Analysis and Visualization Popular 🔥 Data Science R: Data Analysis and Visualization
Advanced
Data Science Python: Machine Learning Popular 🔥 Data Science R: Machine Learning Designing and Implementing Production MLOps New 🎉 Natural Language Processing for Production (NLP) New 🎉
Find Inspiration
Get Course Recommendation Must Try 💎 An Ultimate Guide to Become a Data Scientist
For Companies
For Companies
Corporate Offerings Hiring Partners Candidate Portfolio Hire Our Graduates
Students Work
Students Work
All Posts Capstone Data Visualization Machine Learning Python Projects R Projects
Tutorials
About
About
About Us Accreditation Contact Us Join Us FAQ Webinars Subscription An Ultimate Guide to
Become a Data Scientist
    Login
NYC Data Science Acedemy
Bootcamps
Courses
Students Work
About
Bootcamps
Bootcamps
Data Science with Machine Learning Flagship
Data Analytics Bootcamp
Artificial Intelligence Bootcamp New Release 🎉
Free Lessons
Intro to Data Science New Release 🎉
Find Inspiration
Find Alumni with Similar Background
Job Outlook
Occupational Outlook
Graduate Outcomes Must See 🔥
Alumni
Success Stories
Testimonials
Alumni Directory
Alumni Exclusive Study Program
Courses
Bundles
financing available
View All Bundles
Bootcamp Prep
Data Science Mastery
Data Science Launchpad with Python NEW!
View AI Courses
Generative AI for Everyone
Generative AI for Finance
Generative AI for Marketing
View Data Science Courses
View All Professional Development Courses
Beginner
Introductory Python
Intermediate
Python: Data Analysis and Visualization
R: Data Analysis and Visualization
Advanced
Python: Machine Learning
R: Machine Learning
Designing and Implementing Production MLOps
Natural Language Processing for Production (NLP)
For Companies
Corporate Offerings
Hiring Partners
Candidate Portfolio
Hire Our Graduates
Students Work
All Posts
Capstone
Data Visualization
Machine Learning
Python Projects
R Projects
About
Accreditation
About Us
Contact Us
Join Us
FAQ
Webinars
Subscription
An Ultimate Guide to Become a Data Scientist
Tutorials
Data Analytics
  • Learn Pandas
  • Learn NumPy
  • Learn SciPy
  • Learn Matplotlib
Machine Learning
  • Boosting
  • Random Forest
  • Linear Regression
  • Decision Tree
  • PCA
Interview by Companies
  • JPMC
  • Google
  • Facebook
Artificial Intelligence
  • Learn Generative AI
  • Learn ChatGPT-3.5
  • Learn ChatGPT-4
  • Learn Google Bard
Coding
  • Learn Python
  • Learn SQL
  • Learn MySQL
  • Learn NoSQL
  • Learn PySpark
  • Learn PyTorch
Interview Questions
  • Python Hard
  • R Easy
  • R Hard
  • SQL Easy
  • SQL Hard
  • Python Easy
Data Science Blog > Data Science > Building a Titanic Classifier with End-to-End Machine Learning Pipeline

Building a Titanic Classifier with End-to-End Machine Learning Pipeline

Elsa Amores Vera
Posted on Sep 23, 2024

In this blog post, I’ll guide you through the creation of an end-to-end machine learning pipeline using the Titanic dataset. This project was developed as part of a technical interview, focusing on data exploration, preprocessing, model selection, evaluation, and deployment. Additionally, the project ideation includes a fully automated model training and selection pipeline that streamlines the processes of model optimization, training, and comparison. The aim was to build a scalable and reusable pipeline that could be easily deployed using Docker.

Image 1: A blend of the past and the present: The Titanic sails through calm waters, while modern data visualizations and code surround a dedicated data scientist, seamlessly merging historical elegance with the power of today's analytical technology. This image was generated using DALL·E.

Dataset Overview

The Titanic dataset is a well-known benchmark for binary classification problems. It includes information about passengers on the Titanic and their survival status, with the objective of predicting whether a passenger survived based on the following features:

  • PassengerId: Unique ID for each passenger
  • Survived: Whether the passenger survived (0 = No, 1 = Yes)
  • Pclass: Passenger class (1st, 2nd, or 3rd)
  • Name, Sex, Age: Demographic information
  • SibSp, Parch: Number of siblings/spouses and parents/children aboard
  • Ticket, Fare: Ticket number and fare paid
  • Cabin, Embarked: Cabin number and port of embarkation

The dataset contains 891 rows and 12 columns, and is slightly imbalanced, with 60% non-survivors and 40% survivors.

Exploratory Data Analysis (EDA)

Missing Value Imputation

  • Age: About 20% of the data for "Age" were missing. To impute these, I used the median age based on the passenger class. This way, passengers in the same class with similar conditions had comparable age values.
  • Cabin: 77% of the "Cabin" values were missing. Given the large number of missing values, I imputed these with "U" for unknown and later used this information in feature engineering.
  • Embarked: Only <1% for "Embarked" were missing, which I filled using the mode, as it’s the most frequent port of embarkation.

Visualizations

Using histograms, bar charts, box plots, and correlation heatmaps, I analyzed the relationships between various features and survival. Some key insights:

  • Pclass and "Sex" showed strong correlations with survival, with passengers in 1st class and females having higher survival rates.
  • Age distribution indicated that older passengers were more likely to survive.
  • Fare showed a right-skewed distribution, with higher fares generally associated with higher survival rates.

Variable Processing

Encoding Categorical Variables

For features like "Sex" and "Embarked", I applied One-Hot Encoding, converting them into binary numeric format. These features have no intrinsic order, so this method ensured that they were appropriately handled in the modeling process.

Ordinal Encoding

I used ordinal encoding for "Title" and "Deck" features since both have a natural ranking.

  • For "Title" (extracted from passenger names), I mapped social rankings (e.g., Mr., Mrs., Miss) to numerical values based on their importance.
  • The "Deck" feature, derived from the first letter of the Cabin number, was encoded to reflect different ship levels. Deck A was ranked higher than Deck G, which was closer to the lower decks of the ship.

To ensure robustness in the event of unseen categories during inference, I implemented a mechanism that assigns new categories to a default value of zero, preventing errors during model deployment.

Processing Continuous Variables

The skewness of continuous features like "Age" and "Fare" was assessed to check for asymmetry in their distribution.

  • "Age" had a slight right-skew with a skewness value of 0.54, so I left it untransformed.
  • "Fare" had a skewness of 4.79, indicating a significant right skew. To handle this, I applied a log1p transformation, which normalizes the data and reduces the impact of extreme values. This transformation was preferred over the standard log due to the presence of zero fares.

Feature Engineering

To improve model performance, I created several new features:

  • FamilySize: Combined the "SibSp" and "Parch" features to capture the total number of family members aboard.
  • IsAlone: A binary feature created to indicate whether a passenger was traveling alone.
  • Title: Extracted from the "Name" feature to categorize passengers based on their social title.
  • Deck: Derived from the "Cabin" feature to represent the deck level.

Feature Selection

I used a Random Forest classifier to perform feature selection. This method ranks features based on their importance in improving the model’s performance. After analyzing the feature importance and correlation plots, I dropped several columns, including "SibSp", "Parch" and "IsAlone" as they either had low importance and/or caused multicollinearity. This reduced overfitting and simplified the model.

Model Selection and Training Pipeline

I developed a model training and selection pipeline designed to automate key processes, including model optimization, training, and evaluation. The pipeline efficiently handles hyperparameter tuning through random search and generates comprehensive performance summaries. Additionally, the pipeline automatically saved two versions of each model: one trained only on the train set for model selection, and another trained on the full dataset for inference.

I tested multiple models—Logistic Regression, Random Forest, and XGBoost—using cross-validation to evaluate accuracy, precision, and recall. Given the dataset's slight imbalance (60% non-survivors and 40% survivors), I emphasized metrics beyond accuracy, focusing on precision and recall to capture the model's performance on the minority class (survivors). Random Forest demonstrated the best performance, achieving a cross-validation accuracy of 84.3% along with balanced precision and recall. 

The dataset was split into 70% training data and 30% test data for robust model evaluation. Model selection was based solely on performance metrics from the train set, and only the selected Random Forest model was evaluated on the test set to prevent data leakage and ensure unbiased evaluation.

Evaluation

The final Random Forest model was evaluated on the test set to assess generalization. Given the slight class imbalance, I used additional metrics like precision, recall, and F1-score to evaluate the model’s performance on the minority class (survivors). I also plotted the ROC AUC curve to assess model performance across different thresholds. Using Youden's J statistic, I determined the best threshold to balance false positives and true positives. This threshold was saved and will be used during inference to ensure consistency.

Production Pipeline and Docker Integration

The entire production pipeline was designed to automate data preprocessing, model training, and inference. Key components include:

  • Data Preprocessing Scripts: These scripts handle feature transformations, imputations, and clean data preparation for the model.
  • Configuration: Hardcoded values such as the best threshold, model paths, and final model features are stored in the config folder for easy access during execution.
  • Model and Ideation Code: The optimized models and notebooks containing exploratory analysis and inference logic are saved in separate directories for easy reference.
  • Unit Testing: The tests folder includes unit tests for the pipeline’s key functions. These tests ensure the pipeline operates as expected at every stage.
  • Shell Script and Dockerization: The run_titanic_pipeline_and_tests.sh script automates the pipeline’s execution. The project is fully containerized using Docker, enabling seamless deployment across environments.

Project Execution

This project is fully containerized using Docker, ensuring that all dependencies and configurations are consistently managed across environments. You can initiate the Jupyter Notebook and run the pipeline and unit tests effortlessly. Docker allows for a seamless setup, ensuring that the project remains reproducible and scalable.

Conclusion

This project demonstrates the end-to-end process of building, evaluating, and deploying a machine learning model using the Titanic dataset. The integration with Docker ensures the pipeline is scalable and deployable across various environments without dependency issues. By focusing on feature engineering, robust evaluation techniques, and appropriate metric selection, the model performs effectively, even when facing a slightly imbalanced dataset.

For more details, explore the complete project and production pipeline in the GitHub repository.

About Author

Elsa Amores Vera

Welcome to my blog! I'm Elsa Amores Vera, a seasoned Data Scientist and AI/ML Engineer with a passion for leveraging data to drive impactful change. With a PhD in Biochemistry, Molecular Biology, and Biomedicine, my journey began in...
View all posts by Elsa Amores Vera >

Related Articles

Capstone
Predicting the Unpredictable: Revolutionizing E-commerce Delivery with Machine Learning
Capstone
The Convenience Factor: How Grocery Stores Impact Property Values
Capstone
Acquisition Due Dilligence Automation for Smaller Firms
R Shiny
Forecasting NY State Tax Credits: R Shiny App for Businesses
Machine Learning
Pandemic Effects on the Ames Housing Market and Lifestyle

Leave a Comment

No comments found.

View Posts by Categories

All Posts 2399 posts
AI 7 posts
AI Agent 2 posts
AI-based hotel recommendation 1 posts
AIForGood 1 posts
Alumni 60 posts
Animated Maps 1 posts
APIs 41 posts
Artificial Intelligence 2 posts
Artificial Intelligence 2 posts
AWS 13 posts
Banking 1 posts
Big Data 50 posts
Branch Analysis 1 posts
Capstone 206 posts
Career Education 7 posts
CLIP 1 posts
Community 72 posts
Congestion Zone 1 posts
Content Recommendation 1 posts
Cosine SImilarity 1 posts
Data Analysis 5 posts
Data Engineering 1 posts
Data Engineering 3 posts
Data Science 7 posts
Data Science News and Sharing 73 posts
Data Visualization 324 posts
Events 5 posts
Featured 37 posts
Function calling 1 posts
FutureTech 1 posts
Generative AI 5 posts
Hadoop 13 posts
Image Classification 1 posts
Innovation 2 posts
Kmeans Cluster 1 posts
LLM 6 posts
Machine Learning 364 posts
Marketing 1 posts
Meetup 144 posts
MLOPs 1 posts
Model Deployment 1 posts
Nagamas69 1 posts
NLP 1 posts
OpenAI 5 posts
OpenNYC Data 1 posts
pySpark 1 posts
Python 16 posts
Python 458 posts
Python data analysis 4 posts
Python Shiny 2 posts
R 404 posts
R Data Analysis 1 posts
R Shiny 560 posts
R Visualization 445 posts
RAG 1 posts
RoBERTa 1 posts
semantic rearch 2 posts
Spark 17 posts
SQL 1 posts
Streamlit 2 posts
Student Works 1687 posts
Tableau 12 posts
TensorFlow 3 posts
Traffic 1 posts
User Preference Modeling 1 posts
Vector database 2 posts
Web Scraping 483 posts
wukong138 1 posts

Our Recent Popular Posts

AI 4 AI: ChatGPT Unifies My Blog Posts
by Vinod Chugani
Dec 18, 2022
Meet Your Machine Learning Mentors: Kyle Gallatin
by Vivian Zhang
Nov 4, 2020
NICU Admissions and CCHD: Predicting Based on Data Analysis
by Paul Lee, Aron Berke, Bee Kim, Bettina Meier and Ira Villar
Jan 7, 2020

View Posts by Tags

#python #trainwithnycdsa 2019 2020 Revenue 3-points agriculture air quality airbnb airline alcohol Alex Baransky algorithm alumni Alumni Interview Alumni Reviews Alumni Spotlight alumni story Alumnus ames dataset ames housing dataset apartment rent API Application artist aws bank loans beautiful soup Best Bootcamp Best Data Science 2019 Best Data Science Bootcamp Best Data Science Bootcamp 2020 Best Ranked Big Data Book Launch Book-Signing bootcamp Bootcamp Alumni Bootcamp Prep boston safety Bundles cake recipe California Cancer Research capstone car price Career Career Day ChatGPT citibike classic cars classpass clustering Coding Course Demo Course Report covid 19 credit credit card crime frequency crops D3.js data data analysis Data Analyst data analytics data for tripadvisor reviews data science Data Science Academy Data Science Bootcamp Data science jobs Data Science Reviews Data Scientist Data Scientist Jobs data visualization database Deep Learning Demo Day Discount disney dplyr drug data e-commerce economy employee employee burnout employer networking environment feature engineering Finance Financial Data Science fitness studio Flask flight delay football gbm Get Hired ggplot2 googleVis H20 Hadoop hallmark holiday movie happiness healthcare frauds higgs boson Hiring hiring partner events Hiring Partners hotels housing housing data housing predictions housing price hy-vee Income industry Industry Experts Injuries Instructor Blog Instructor Interview insurance italki Job Job Placement Jobs Jon Krohn JP Morgan Chase Kaggle Kickstarter las vegas airport lasso regression Lead Data Scienctist Lead Data Scientist leaflet league linear regression Logistic Regression machine learning Maps market matplotlib Medical Research Meet the team meetup methal health miami beach movie music Napoli NBA netflix Networking neural network Neural networks New Courses NHL nlp NYC NYC Data Science nyc data science academy NYC Open Data nyc property NYCDSA NYCDSA Alumni Online Online Bootcamp Online Training Open Data painter pandas Part-time performance phoenix pollutants Portfolio Development precision measurement prediction Prework Programming public safety PwC python Python Data Analysis python machine learning python scrapy python web scraping python webscraping Python Workshop R R Data Analysis R language R Programming R Shiny r studio R Visualization R Workshop R-bloggers random forest Ranking recommendation recommendation system regression Remote remote data science bootcamp Scrapy scrapy visualization seaborn seafood type Selenium sentiment analysis sentiment classification Shiny Shiny Dashboard Spark Special Special Summer Sports statistics streaming Student Interview Student Showcase SVM Switchup Tableau teachers team team performance TensorFlow Testimonial tf-idf Top Data Science Bootcamp Top manufacturing companies Transfers tweets twitter videos visualization wallstreet wallstreetbets web scraping Weekend Course What to expect whiskey whiskeyadvocate wildfire word cloud word2vec XGBoost yelp youtube trending ZORI

NYC Data Science Academy

NYC Data Science Academy teaches data science, trains companies and their employees to better profit from data, excels at big data project consulting, and connects trained Data Scientists to our industry.

NYC Data Science Academy is licensed by New York State Education Department.

Get detailed curriculum information about our
amazing bootcamp!

Please enter a valid email address
Sign up completed. Thank you!

Offerings

  • HOME
  • DATA SCIENCE BOOTCAMP
  • ONLINE DATA SCIENCE BOOTCAMP
  • Professional Development Courses
  • CORPORATE OFFERINGS
  • HIRING PARTNERS
  • About

  • About Us
  • Alumni
  • Blog
  • FAQ
  • Contact Us
  • Refund Policy
  • Join Us
  • SOCIAL MEDIA

    © 2025 NYC Data Science Academy
    All rights reserved. | Site Map
    Privacy Policy | Terms of Service
    Bootcamp Application