jer233 commited on
Commit
2b33c6f
·
verified ·
1 Parent(s): 00d8fb5

Create regression_model_loader.py

Browse files
Files changed (1) hide show
  1. regression_model_loader.py +14 -0
regression_model_loader.py ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import joblib
2
+ from utils import DEVICE
3
+
4
+
5
+ class RegressionModelLoader:
6
+ def __init__(self):
7
+ print("Regression Model init")
8
+ self.model = self.load("./logistic_regression_model.pkl")
9
+
10
+ def load(self, regression_model_file_name):
11
+ return joblib.load(regression_model_file_name)
12
+
13
+
14
+ regression_model = RegressionModelLoader()