Decision tree for titanic dataset in Python

 Build decision tree model to predict survival based on certain parameters :




import library


Read data set using panda library's read_csv method.


 
In this file using following columns build a model to predict if person would survive or not 1.Pclass 2.Sex 3.Age 4.Fare

so we will drop unnecessary columns :


 
Now we separate dependent and independent   data frame for pass into our model



Gender column contain categorical data so we have too convert it into numerical for this we will use map method of pandas. and also it contain missing value which we will handle using fillna method.


 

Now train test split :




Building Decision tree model for titanic dataset in Python



Confusion matrix :



if you want to download data set then click here.


Comments

Post a Comment

Popular posts from this blog

Linear Regression

Image
  Linear Regression Using SKlearn We will predict the hosing price here by creating Machine Learning's Linear Regression Model. let's start : First we import the necessary library  Now we import the data from csv file. You can import csv file in python using pandas library's read_csv  method. lets visualise the data using matplotlib library we have to supply X & Y variable  means independent and dependent variable in leaner model so here independent variable is area and we will here predict price of home using area so price is depend on area size so price is dependent variable. you can drop column in pandas using drop   method . here x = new_df & y = df.price Now we build a leaner regression model  here we have to pass our x & y in fit method which will train our model. Lets predict the price of house which have 5000 sqft area. Predict method always accept 2D Array thats why we use double bracket here. price of 5000 sqft area house is 85955...
Read more

K Means Cluster Algorithm

Image
 K Means Cluster  Its a distance based algorithm. The k-means algorithm is a  clustering algorithm . That means that you have a bunch of points in some space, and you want to guess what groups they seem to be in.  K-Means cluster in Python : First of all import necessary  library and read the csv file using panda library Lets visualize data    Now we build the KMeans  cluster Now we find the center of group  You can clearly saw that center is not perfectly fit in group because we didn't performing scaling  on over data set. so first we do the scaling on over data set. Now again we build KMeans cluster model Now we again find center of our data groups Now question is How we know the value of K ?  We can find the value of K using elbow method. In this method we plot the data and where you find the major difference in graph that point is your value of K. You can download data set by click here.
Read more