ML-Monitoring/handler.py

import datetime as dt
import yfinance as yf
import dagshub
import mlflow

def get_experiment_id(name):
    exp = mlflow.get_experiment_by_name(name)
    if exp is None:
        exp_id = mlflow.create_experiment(name)
        return exp_id
    return exp.experiment_id

def load_data():
    Stock_Number= "BTC-USD"
    End_date = "2022-02-08"
    Period = 5
    Interval = 1
    
    enddate = dt.datetime.strptime(End_date, "%Y-%m-%d").date()
    startdate = enddate - dt.timedelta(days=365 * Period) # for 5 years
    data = yf.download(tickers=Stock_Number, start=startdate, end=enddate, interval=f"{Interval}d")

    return data

if __name__ == "__main__":
    pass
    # data = load_data()
    # dagshub.auth.get_token()