From 2111b8623ad892b885d6345d54b5f3cfa94d6320 Mon Sep 17 00:00:00 2001 From: Alexandre Gattiker Date: Tue, 4 Feb 2020 13:43:01 +0100 Subject: [PATCH] Update train.py --- diabetes_regression/training/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/diabetes_regression/training/train.py b/diabetes_regression/training/train.py index f56daa99..a145a843 100644 --- a/diabetes_regression/training/train.py +++ b/diabetes_regression/training/train.py @@ -31,7 +31,7 @@ from sklearn.linear_model import Ridge from sklearn.metrics import mean_squared_error from sklearn.model_selection import train_test_split -from sklearn.externals import joblib +import joblib def train_model(run, data, alpha):