Remove load_boston in survival.py

stanfordmlgroup · Nov 6, 2023 · 1710840 · 1710840
1 parent 0154ac5
commit 1710840
Showing 1 changed file with 6 additions and 2 deletions.
diff --git 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