Sign In
Learn practical skills, build real-world projects, and advance your career

Learn how to predict survival on the Titanic using Deep Learning with PyTorch. Import necessary modules, clean data, and analyze features for accurate predictions.

TITANIC Prediction

This project is build by Suryansh Shrivastava for Course Project of "Deep Learning with PyTorch: Zero to GANs" of

# Importing Necessary Modules
import numpy as np
import pandas as pd       
import tensorflow.compat.v1 as tf
from sklearn.model_selection import train_test_split
import matplotlib.pyplot as plt
import seaborn as sns
import jovian 

Reading and cleaning the input data

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')