Skip to content

Instantly share code, notes, and snippets.

@oneryalcin

oneryalcin/sparkify_9_inexer_onehotencoder.py

Created September 23, 2019 23:04
Show Gist options
  • Save oneryalcin/ee3a7adf7a7d593b197eeb0a31f96a19 to your computer and use it in GitHub Desktop.
Save oneryalcin/ee3a7adf7a7d593b197eeb0a31f96a19 to your computer and use it in GitHub Desktop.
Download ZIP
9 StringIndexer and OneHotEncoder
Raw
sparkify_9_inexer_onehotencoder.py
joined = StringIndexer(inputCol='gender', outputCol='gender_idx')\
.fit(joined)\
.transform(joined)
joined = StringIndexer(inputCol='level', outputCol='level_idx')\
.fit(joined)\
.transform(joined)
joined = OneHotEncoderEstimator(inputCols=['gender_idx', 'level_idx'],
outputCols=['gender_dummy','level_dummy'])\
.fit(joined)\
.transform(joined)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment