/kaggle-titanic-deeplearn-js

Kaggle's Titanic with deeplearn.js

Primary LanguageTypeScriptMIT LicenseMIT

Kaggle's Titanic with deeplearn.js

Demo

yarn install
yarn build

Open ./index.html and the browser console.

Check the GPU process.

Data Handling

  • Remove not useful data (PassengerId, Name, Ticket, Cabin)
  • Replace missing Age with average
  • Feature scaling (Age, SibSp, Parch, Fare)
  • Categorical data to numerical (Sex, Embarked)

Model

  • Multilayer Perceptron
  • SGD