utils/fetcher.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on 2024-07-19 20:25:49 Friday

@author: Nikhil Kapila
"""

import torch, pickle

def model_folder_name(ft, type):
    return f'bgd_{type}_{ft}_LSTM'

def fetch_model(ft, type, dir='metaflow_models_30hrs'):
    file = f'{dir}/{model_folder_name(ft,type)}/{model_folder_name(ft,type)}.pth'
    print(file)
    return torch.load(file)

def fetch_scaler(ft, type, dir='metaflow_models_30hrs'):
    scaler_dir = f'{dir}/{model_folder_name(ft,type)}/scaler_{model_folder_name(ft, type)}.pkl'
    print(scaler_dir)
    with open(scaler_dir, 'rb') as file: scaler = pickle.load(file)
    return scaler