Skip to content

Instantly share code, notes, and snippets.

@gvyshnya
Created April 18, 2021 20:01
Show Gist options
  • Save gvyshnya/5352164caf6dfc1f66efcf5efae23a85 to your computer and use it in GitHub Desktop.
Save gvyshnya/5352164caf6dfc1f66efcf5efae23a85 to your computer and use it in GitHub Desktop.
Featurewiz Snippet
from featurewiz import featurewiz
# out1 and out2: is always a tuple containing two objects.
# In our case, it is going to be "test" and "train" dataframes with the important features selected by featurewiz
trainm, testm = featurewiz(
df_train,
target='target',
corr_limit=0.7,
verbose=2,
sep=",",
header=0,
test_data=df_test,
feature_engg="", # we do not want featurewiz to create any additional features in this experiment
category_encoders="OrdinalEncoder" #OrdinalEncoder
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment