PyCaret Sample

pycaret_sample.py

import comet_ml

from pycaret.datasets import get_data
from pycaret.classification import *

comet_ml.init(project_name="pycaret")

dataset = get_data("titanic")
clf1 = setup(
    data=dataset,
    target="Survived",
    log_experiment="comet_ml",
    experiment_name="pycaret_example",
    log_data=True,
)
model = compare_models(sort="Accuracy")
predict_model(model)
save_model(model, "titanic")

comet_logger = clf1[0].logging_param.loggers[0]
comet_logger.finish_experiment()
comet_logger.run.display("panels")

Pretty neat tool