Skip to content

This project focuses on the analysis of the Titanic dataset, aiming to predict survival outcomes of passengers using a machine learning model (with explanations, using LIME)

Notifications You must be signed in to change notification settings

SonnyFixit/Machine_Learning_LIME

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Machine Learning - LIME

Content

General info

The Titanic Survival Prediction Project is a machine learning endeavor aimed at developing a predictive model to determine the survival outcomes of passengers aboard the RMS Titanic. Leveraging historical passenger data, this project applies data preprocessing techniques, utilizes a Decision Tree Classifier for predictions, and employs the Local Interpretable Model-agnostic Explanations (LIME) framework to interpret the model's decisions.

Key points of the program

  • Data Preprocessing
  • Model Training and Evaluation
  • Interpretability with LIME

Technologies

  • Python 3.11.3
  • Visual Studio Code
  • Seaborn and Matplotlib (for data visualization)
  • Pandas (for data manipulation)
  • LIME (for model interpretability)
  • Scikit-learn (for model training and evaluation)

Features

  • Comprehensive data preprocessing to handle missing values and convert categorical data
  • Decision Tree Classifier to predict survival outcomes
  • Evaluation of model performance using classification metrics and confusion matrix
  • Interpretation of individual predictions with LIME

Team:

About

This project focuses on the analysis of the Titanic dataset, aiming to predict survival outcomes of passengers using a machine learning model (with explanations, using LIME)

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages