From 17108402129fcc98ffc2bf65877b869f6c1a8888 Mon Sep 17 00:00:00 2001
From: Yixing <108294040+jyx-su@users.noreply.github.com>
Date: Mon, 6 Nov 2023 13:01:21 -0800
Subject: [PATCH] Remove load_boston in survival.py

---
 examples/survival.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/examples/survival.py b/examples/survival.py
index c830db2..90037a2 100644
--- a/examples/survival.py
+++ b/examples/survival.py
@@ -1,5 +1,4 @@
 import numpy as np
-from sklearn.datasets import load_boston
 from sklearn.metrics import mean_squared_error
 from sklearn.model_selection import train_test_split
 
@@ -8,7 +7,12 @@
 
 if __name__ == "__main__":
 
-    X, Y = load_boston(return_X_y=True)
+    #Load Boston housing dataset
+    data_url = "http://lib.stat.cmu.edu/datasets/boston"
+    raw_df = pd.read_csv(data_url, sep="\s+", skiprows=22, header=None)
+    X = np.hstack([raw_df.values[::2, :], raw_df.values[1::2, :2]])
+    Y = raw_df.values[1::2, 2]
+    
     X_train, X_test, Y_train, Y_test = train_test_split(X, Y, test_size=0.2)
 
     # introduce administrative censoring