wncc / Hello-FOSS-ML

The Machine Learning repository for Hello-FOSS
5 stars 31 forks source link

Titanic - Hyperparameter tuning #10

Open Aakriti28 opened 4 years ago

Aakriti28 commented 4 years ago

A neural network has been trained on the dataset in this notebook but it has very low accuracy. Your task is to tune the hyperparameters and improve the model. Some suggestions - modifying the optimizer, adding more hidden layers to the model or changing the dimension of layers, adding dropout, regularization etc.