Skip to content

Instantly share code, notes, and snippets.

@WillKoehrsen
Created October 3, 2018 14:23
Show Gist options
  • Save WillKoehrsen/2226f2558050a098aea34124abcbe160 to your computer and use it in GitHub Desktop.
Save WillKoehrsen/2226f2558050a098aea34124abcbe160 to your computer and use it in GitHub Desktop.
import featuretools as ft
import featuretools.variable_types as vtypes
# Create entityset
es = ft.EntitySet(id = 'customers')
# Create entity from members
es.entity_from_dataframe(entity_id='members', dataframe=members,
index = 'msno',
time_index = 'registration_init_time',
variable_types = {'city': vtypes.Categorical,
'registered_via': vtypes.Categorical})
# Add other entities ...
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment