Skip to content

Instantly share code, notes, and snippets.

@kvnkho
Created January 6, 2022 18:46
Show Gist options
  • Save kvnkho/b227b81d394cfdbd1ef7152617df2a8c to your computer and use it in GitHub Desktop.
Save kvnkho/b227b81d394cfdbd1ef7152617df2a8c to your computer and use it in GitHub Desktop.
from pycaret.classification import *
clf = setup(data = df,
target = "Survived",
session_id = 123,
silent = True,
verbose = False,
html = False)
models = compare_models(fold = 5,
round = 4,
sort = "Accuracy",
turbo = True,
n_select = 5,
verbose = False)
results = pull().reset_index(drop = True)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment