diff --git a/examples/regression.py b/examples/regression.py index 88122cd..971a7b2 100644 --- a/examples/regression.py +++ b/examples/regression.py @@ -1,8 +1,7 @@ -from sklearn.metrics import mean_squared_error -from sklearn.model_selection import train_test_split - import numpy as np import pandas as pd +from sklearn.metrics import mean_squared_error +from sklearn.model_selection import train_test_split from ngboost import NGBRegressor from ngboost.distns import Normal diff --git a/examples/survival.py b/examples/survival.py index db59caf..32be7a5 100644 --- a/examples/survival.py +++ b/examples/survival.py @@ -1,6 +1,5 @@ import numpy as np import pandas as pd - from sklearn.metrics import mean_squared_error from sklearn.model_selection import train_test_split