Skip to content

Instantly share code, notes, and snippets.

@VincentGat

VincentGat/model_features.py

Created February 24, 2020 16:46
Show Gist options
  • Save VincentGat/fb68ec6cfb5f18335150d0564f22cf1c to your computer and use it in GitHub Desktop.
Save VincentGat/fb68ec6cfb5f18335150d0564f22cf1c to your computer and use it in GitHub Desktop.
Download ZIP
Raw
model_features.py
model_family = random.choice(['nb', 'rf', 'lgbm', 'catboost'], p=[0, 0.3334, 0.3333, 0.3333])
features = random.choice([True, False], nb_features)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment