From ad7157da2d8b15b117e0de853dfc49630f56cf64 Mon Sep 17 00:00:00 2001
From: Aman Sharma <76823502+arcAman07@users.noreply.github.com>
Date: Mon, 21 Feb 2022 17:03:38 +0530
Subject: [PATCH] Create titanic.jl

---
 other/Titanic/titanic.jl | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)
 create mode 100644 other/Titanic/titanic.jl

diff --git a/other/Titanic/titanic.jl b/other/Titanic/titanic.jl
new file mode 100644
index 000000000..34083d1a2
--- /dev/null
+++ b/other/Titanic/titanic.jl
@@ -0,0 +1,21 @@
+using MLDatasets:Titanic
+
+function get_processed_data(args)
+    labels = Titanic.targets()
+    features = Titanic.features()
+
+    # Split into training and test sets, 2/3 for training, 1/3 for test.
+    train_indices = [1:3:891; 2:3:891]
+
+    X_train = features[:, train_indices]
+    y_train = labels[:, train_indices]
+
+    X_test = normed_features[:, 3:3:891]
+    y_test = onehot_labels[:, 3:3:891]
+
+    #repeat the data `args.repeat` times
+    train_data = Iterators.repeated((X_train, y_train), args.repeat)
+    test_data = (X_test, y_test)
+
+    return train_data, test_data
+end