Learn how to predict survival on the Titanic using Deep Learning with PyTorch. Import necessary modules, clean data, and analyze features for accurate predictions.
This project is build by Suryansh Shrivastava for Course Project of "Deep Learning with PyTorch: Zero to GANs" of Jovian.ai
# Importing Necessary Modules
import numpy as np
import pandas as pd
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()
from sklearn.model_selection import train_test_split
import matplotlib.pyplot as plt
import seaborn as sns
import jovian
We first read the CSV input file using Pandas. Next, we remove irrelevant entries, and prepare the data for our neural network.
# Read the CSV input file and show first 5 rows
df_train = pd.read_csv('../input/train.csv')
df_train.head(5)