Logistic Regression in ML and AI with Examples and Projects The Complete Code import numpy as np import pandas as pd import matplotlib.pyplot as plt import seaborn as sns from sklearn.model_selection import train_test_split train = pd.read_csv("titanic_train.csv") def impute_age(cols): Age = cols[0] Pclass = cols[1] if pd.isnull(Age): if Pclass == 1: return 37 elif Pclass == 2: return 29 else: return 24 else: return Age train['Age'] = train[['Age','Pclass']].apply(impute_age,axis=1) train.drop('Cabin',axis=1,inplace=True) train.dropna(inplace=True) Sex = pd.get_dummies(train['Sex'],drop_first=True) Embark = pd.get_dummies(train['Embarked'],drop_first=True) train = pd.concat([train,Sex,Em...
Comments
Post a Comment