Skip to content

A complete machine learning pipeline using Decision Tree Regression with preprocessing transformations. The model is serialized using Pickle, enabling easy reuse for predictions on new data.

Notifications You must be signed in to change notification settings

fahadabid545/Creating-Pipeline

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

52 Commits
Β 
Β 
Β 
Β 

Repository files navigation

🌊 Titanic Dataset - Machine Learning Pipeline πŸ› οΈ

✨ Project Overview

This project demonstrates how to build a Machine Learning Pipeline for the Titanic Dataset using Decision Tree Classifier. The pipeline includes:

  • Data Preprocessing (handling missing values, encoding, and scaling)
  • Feature Selection using SelectKBest
  • Model Training with Decision Tree Classifier
  • Pipeline Serialization using Pickle

πŸ“Š Data Preprocessing

  • Handling Missing Values: SimpleImputer()
  • Encoding Categorical Features: OneHotEncoder()
  • Feature Scaling: MinMaxScaler()
  • Feature Selection: SelectKBest(score_func=chi2, k=8)

🧠 Model Training

  • Algorithm: DecisionTreeClassifier()
  • Pipeline: Created with ColumnTransformer and Pipeline
  • Serialization: Saved using Pickle

About

A complete machine learning pipeline using Decision Tree Regression with preprocessing transformations. The model is serialized using Pickle, enabling easy reuse for predictions on new data.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published