Titanic: Code for a Kaggle Submission

.libPaths("/home/rstudioshared/shared_files/packages")
library(dplyr)
library(ggplot2)

titanic <- read.csv("/home/rstudioshared/shared_files/data/titanic_train.csv")
titanic_test <- read.csv("/home/rstudioshared/shared_files/data/titanic_test.csv")


train_set_summary <- titanic %>% mutate(age.group = cut(Age, breaks=seq(0,90,10))) %>% 
       group_by(Sex,age.group, Pclass) %>%
       summarize(num_passengers=n(),SurvivalRate = mean(Survived)) 

nrow(titanic)
nrow(titanic_test)



titanic_test_join <- left_join(titanic_test %>% mutate(age.group = cut(Age, breaks=seq(0,90,10))), 
          train_set_summary, by=c("Sex", "age.group", "Pclass"))



titanic_test_join %>% 
  mutate(Survived = round(SurvivalRate)) %>% 
  mutate(Survived = ifelse(is.na(Survived), 0, Survived)) %>%
  select(PassengerId, Survived) %>% 
  write.csv("kaggle_preds_Sex_Age_Pclass.csv", row.names=FALSE)