pr5 AAM
pr5 AAM
Roll No : 56
Class : TYAN
import pandas as pd
import numpy as np
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import LabelEncoder
from sklearn.tree import DecisionTreeClassifier
from sklearn.metrics import accuracy_score
df=pd.read_csv("/content/drive/MyDrive/Decision_tree.csv")
print(df.head())
if "Sr.No" in df.columns:
df = df.drop(columns=["Sr.No"])
label_encoders = {}
for col in df.columns:
le = LabelEncoder()
df[col] = le.fit_transform(df[col])
label_encoders[col] = le
X = df.drop(columns=["Play"])
y = df["Play"]
clf = DecisionTreeClassifier()
clf.fit(X_train, y_train)
y_pred = clf.predict(X_test)
weather_predictions