Scribd
Upload
24 views

Assignment - 4 - Decision Tree - 014319

Decision tree ml

Uploaded by

vedikadhebe2712
Copyright
© © All Rights Reserved
We take content rights seriously. If you suspect this is your content, claim it here.
Available Formats
Download as PDF, TXT or read online on Scribd
24 views

Assignment - 4 - Decision Tree - 014319

Decision tree ml

Uploaded by

vedikadhebe2712
Copyright
© © All Rights Reserved
We take content rights seriously. If you suspect this is your content, claim it here.
Available Formats
Download as PDF, TXT or read online on Scribd
You are on page 1/ 3

Assignment No.

Roll no. - 33309

[2]: import numpy as np


import pandas as pd

[3]: dataset = pd.read_csv("admission_data.csv")

[7]:

[7]: Serial No. GRE Score TOEFL Score University Rating SOP LOR CGPA \
0 1 337 118 4 4.5 4.5 9.65
1 2 324 107 4 4.0 4.5 8.87
2 3 316 104 3 3.0 3.5 8.00
3 4 322 110 3 3.5 2.5 8.67
4 5 314 103 2 2.0 3.0 8.21

Research Chance of Admit


0 1 1.0
1 1 0.0
2 1 0.0
3 1 1.0
4 0 0.0

[8]:

[8] : Serial No. 0


GRE Score 0
TOEFL Score 0
University Rating 0
SOP 0
LOR 0
CGPA 0
Research 0
Chance of Admit 0
dtype: int64

[9] : dataset.columns

1
[9] : Index(['Serial No.', 'GRE Score', 'TOEFL Score', 'University Rating', 'SOP',
'LOR ', 'CGPA', 'Research', 'Chance of Admit '],
dtype='object')

[10] : dataset.loc[dataset['Chance of Admit < 0.8, 'Chance of Admit = 0


dataset.loc[dataset['Chance of Admit >= 0.8, 'Chance of Admit =1

[11] : X = Admit =
y= Admit

[12] : from sklearn.model_selection import train_test_split


=
‹→random_state = 123)

[15]: from sklearn.tree import DecisionTreeClassifier, plot_tree


from sklearn import metrics

dtc = DecisionTreeClassifier()

dtc = dtc.fit(X_train, Y_train)

y_pred = dtc.predict(X_test)

print("Confusion Matrix: \n")


print(metrics.confusion_matrix(Y_test, y_pred))
print("\n1. Accuracy Score:\t", metrics.accuracy_score(Y_test, y_pred))
print("2. Precision Score:\t",metrics.precision_score(Y_test, y_pred))
print("3. Recall Score:\t", metrics.recall_score(Y_test, y_pred))
print("4. f1 Score:\t", metrics.f1_score(Y_test, y_pred))

Confusion Matrix:

[[79 3]
[ 4 39]]

1. Accuracy Score: 0.944


2. Precision Score: 0.9285714285714286
3. Recall Score: 0.9069767441860465
4. f1 Score: 0.9176470588235294

[19]:
plot_tree(dtc, feature_names=X.columns, class_names=['0', '1'], filled=True)

2
3

You might also like