From 428d9e9e2a7cd5ead3ece6d2b2522874fae6b66b Mon Sep 17 00:00:00 2001 From: jyx-su <108294040+jyx-su@users.noreply.github.com> Date: Wed, 8 Nov 2023 13:21:05 -0800 Subject: [PATCH] black format --- examples/regression.py | 5 ++--- examples/survival.py | 5 ++--- 2 files changed, 4 insertions(+), 6 deletions(-) diff --git a/examples/regression.py b/examples/regression.py index 735d47e..88122cd 100644 --- a/examples/regression.py +++ b/examples/regression.py @@ -8,13 +8,12 @@ from ngboost.distns import Normal if __name__ == "__main__": - - #Load Boston housing dataset + # 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) ngb = NGBRegressor(Dist=Normal).fit(X_train, Y_train) diff --git a/examples/survival.py b/examples/survival.py index 4af3ea4..db59caf 100644 --- a/examples/survival.py +++ b/examples/survival.py @@ -8,13 +8,12 @@ from ngboost.distns import LogNormal if __name__ == "__main__": - - #Load Boston housing dataset + # 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