Spaces:

songyiliao
/

R-Detect

Sleeping

App Files Files Community

R-Detect / regression_model_loader.py

songyiliao

CoinW

feat: initial cmommit (#1)

1244519 verified over 1 year ago

Raw History Blame Contribute Delete

375 Bytes

import joblib
from utils import get_device

DEVICE = get_device()


class RegressionModelLoader:
    def __init__(self):
        print("Regression Model init")
        self.model = self.load("./logistic_regression_model.pkl")

    def load(self, regression_model_file_name):
        return joblib.load(regression_model_file_name)


regression_model = RegressionModelLoader()

	import joblib
	from utils import get_device

	DEVICE = get_device()


	class RegressionModelLoader:
	def __init__(self):
	print("Regression Model init")
	self.model = self.load("./logistic_regression_model.pkl")

	def load(self, regression_model_file_name):
	return joblib.load(regression_model_file_name)


	regression_model = RegressionModelLoader()