Skip to content

Instantly share code, notes, and snippets.

@amankharwal

amankharwal/lightgbm.py Secret

Created January 15, 2021 06:30
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save amankharwal/99fbb1b1cdf907b82d4a31df1efc1439 to your computer and use it in GitHub Desktop.
Save amankharwal/99fbb1b1cdf907b82d4a31df1efc1439 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
lightgbm.py
train_data = pd.get_dummies(train_data, columns=["color"], prefix=["color"])
map_type = {"Ghoul":1, "Goblin":2, "Ghost":0} # change ghoul, goblin, ghost to 1, 2 and 0
train_data.loc[:, "type"] = train_data.type.map(map_type)
train_data = train_data.set_index('id')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment