SCAAI_Finance / load_models.py
Mr-Thop's picture
Load Scaler and encoders
0dc5151
from huggingface_hub import hf_hub_download
import joblib
def load_req():
scaler = joblib.load(
hf_hub_download("Mr-Thop/SCAAI", "scaler.joblib")
)
encoder = joblib.load(
hf_hub_download("Mr-Thop/SCAAI", "encoder.joblib")
)
return scaler, encoder
def load_rf():
model = joblib.load(
hf_hub_download("Mr-Thop/SCAAI", "rf.joblib")
)
return model
def load_dt():
model = joblib.load(
hf_hub_download("Mr-Thop/SCAAI", "dt.joblib")
)
return model
def load_nb():
model = joblib.load(
hf_hub_download("Mr-Thop/SCAAI", "nb.joblib")
)
return model