Import sklearn.linear_model for linear regression

RichieHakim · Sep 10, 2024 · ebceeb7 · ebceeb7
1 parent 1b45097
commit ebceeb7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/bnpm/linear_regression.py b/bnpm/linear_regression.py
@@ -2,6 +2,8 @@
 import gc
 
 import numpy as np
+import sklearn
+import sklearn.linear_model
 
 
 def LinearRegression_sweep(X_in,