Skip to content

Instantly share code, notes, and snippets.

@pycaret

pycaret/define_data_types.py

Created July 1, 2020 16:57
Show Gist options
  • Save pycaret/ee1ab553b520101370b4ec22591dcacd to your computer and use it in GitHub Desktop.
Save pycaret/ee1ab553b520101370b4ec22591dcacd to your computer and use it in GitHub Desktop.
Download ZIP
Raw
define_data_types.py
# import regression module
from pycaret.regression import *
# init setup
reg1 = setup(data, target = 'charges', silent=True,
categorical_features=['sex', 'smoker', 'region', 'children'],
numeric_features=['age', 'bmi'])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment