Skip to content

ayush-bhadwal/ML-model

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 

Repository files navigation

Machine Learning Model Comparison (Titanic Dataset)

Project Overview

This project compares multiple Machine Learning models on the Titanic dataset to predict passenger survival.
The goal is to evaluate different algorithms and identify the best-performing model.


Dataset

  • Dataset used: Titanic Dataset (from seaborn)
  • Target variable: survived
  • Features include:
    • pclass, sex, age, sibsp, parch, fare, embarked

Technologies Used

  • Python 🐍
  • NumPy
  • Pandas
  • Matplotlib
  • Seaborn
  • Scikit-learn

Machine Learning Models Used

  • Logistic Regression
  • K-Nearest Neighbors (KNN)
  • Gaussian Naive Bayes
  • Decision Tree Classifier
  • Support Vector Machine (SVM)

Model Performance

Model Accuracy
SVM 0.83
Logistic Regression 0.80
KNN 0.79
Decision Tree ~0.78
Naive Bayes 0.77

Best Model

Support Vector Machine (SVM) achieved the highest accuracy.


Key Steps Performed

  1. Data Loading (Seaborn Titanic Dataset)
  2. Data Cleaning & Preprocessing
  3. Feature Encoding
  4. Feature Scaling
  5. Train-Test Split
  6. Model Training
  7. Model Evaluation (Accuracy, Confusion Matrix, Classification Report)
  8. Model Comparison

Evaluation Metrics

  • Accuracy Score
  • Confusion Matrix
  • Precision, Recall, F1-score
  • Cross-Validation

Project Structure

  • Notebook

About

ML-model comparison project on titanic dataset

Resources

Stars

0 stars

Watchers

0 watching

Forks

Releases

No releases published

Packages

 
 
 

Contributors