# to install pycaret-nightly
# pip install pycaret-nightly

from pycaret.utils import version
version()

from pycaret.datasets import get_data
data = get_data('boston')

from pycaret.regression import *
reg1 = setup(data, target = 'medv', logging=True, experiment_name='boston-tf-meetup-temp')

top5 = compare_models(n_select = 5)

print(top5)

blender = blend_models(estimator_list = top5)

stacker = stack_models(estimator_list = top5[1:], meta_model = top5[0])

tuned_et = tune_model(top5[1], n_iter=50)

!mlflow ui

lightgbm = create_model('lightgbm')

import numpy as np

for i in np.arange(0.1,1,0.01):
    create_model('lightgbm', learning_rate=i, verbose=False)

!mlflow ui