TASK 6: Prediction using Decision Tree Algorithm
GRIP JUNE 2021 Data Science and Business Analytics Internship
Intern : SHUBHAM
IN THIS PROJECT WE ARE EXPLORING A DATASET NAMED AS IRIS.csv IN THIS WE HAVE USED LIBRARIES:
- Pandas
- sklearn.DATASET
- sklearn.TREE
Then we have imported the Dataset as X and got 4 features as columns
- sepal length (cm)
- sepal width (cm)
- petal length (cm)
- petal width (cm)
and explored by some commands like X.head() X.tail() X.info() X.describe() X.isnull().sum()
Then we have: Split dataset into train and test sets
&
from sklearn.model_selection we import train_test_split
Then
Defining the Decision Tree Algorithm and Predicting until the gini value comes to 0.00
And thats how task 6 is completed.