Professional Documents
Culture Documents
Exp 9 - 2131
Exp 9 - 2131
Aravind
VU21CSEN0102131
# Import necessary libraries
import numpy as np
import pandas as pd
from sklearn.datasets import load_iris
from sklearn.tree import DecisionTreeClassifier
from sklearn.model_selection import
train_test_split
from sklearn.metrics import accuracy_score
import dtreeviz
print("___________________________________
________")
# Create a DataFrame from the dataset
df = pd.DataFrame(data.data,
columns=data.feature_names)
df['target'] = y_labels # Add the species labels as
a new column
print(df.head())
print("___________________________________
________")
# Display specific fields for 'setosa' (class 0)
setosa_data = df[df['target'] == 'setosa']
print("Data for 'Setosa' (Class 0):")
print(setosa_data.head(3))
print("___________________________________
________")
# Display specific fields for 'versicolor' (class 1)
versicolor_data = df[df['target'] == 'versicolor']
print("\nData for 'Versicolor' (Class 1):")
print(versicolor_data.head(3))
print("___________________________________
________")
# Display specific fields for 'virginica' (class 2)
virginica_data = df[df['target'] == 'virginica']
print("\nData for 'Virginica' (Class 2):")
print(virginica_data.head(3))
print("___________________________________
________")
# We'll focus on binary classification, e.g.,
classifying 'setosa' (class 0) vs. 'versicolor' (class 1)
#y_binary = (y <= 1).astype(int) # 1 if 'setosa' or
'versicolor', 0 if 'virginica'
m.view()
OUTPUT