Python Experiment
Python Experiment
import pandas as pd
df = sns.load_dataset('titanic')
df['age'].fillna(df['age'].median(), inplace=True)
df['embarked'].fillna(df['embarked'].mode()[0], inplace=True)
label_enc = LabelEncoder()
df['sex'] = label_enc.fit_transform(df['sex'])
df['embarked'] = label_enc.fit_transform(df['embarked'])
X = df.drop('survived', axis=1)
y = df['survived']
# Train-test split
# Naive Bayes
nb_model = GaussianNB()
nb_model.fit(X_train, y_train)
nb_preds = nb_model.predict(X_test)
j48_model = DecisionTreeClassifier(random_state=42)
j48_model.fit(X_train, y_train)
j48_preds = j48_model.predict(X_test)
# KNN Classifier
knn_model = KNeighborsClassifier(n_neighbors=5)
knn_model.fit(X_train, y_train)
knn_preds = knn_model.predict(X_test)
print("Summary of Accuracies:")