cap_backend / utils /load_models.py
logeswari's picture
commit the changes
efcc6b0
raw
history blame contribute delete
556 Bytes
# app/utils/load_models.py
import joblib
import os
models_path = os.path.join(os.path.dirname(__file__), '../models')
# Load models
satisfaction_model = joblib.load(os.path.join(models_path, 'satisfaction_model.pkl'))
performance_model = joblib.load(os.path.join(models_path, 'performance_model.pkl'))
retention_model = joblib.load(os.path.join(models_path, 'retention_model.pkl'))
training_model = joblib.load(os.path.join(models_path, 'training_model.pkl'))
# Load label encoder
label_enc = joblib.load(os.path.join(models_path, 'label_encoder.pkl'))