Skip to content

Instantly share code, notes, and snippets.

@lucasastorian
Last active March 2, 2020 12:17
Show Gist options
  • Select an option

  • Save lucasastorian/e590214482ea094c5a397e791f79e985 to your computer and use it in GitHub Desktop.

Select an option

Save lucasastorian/e590214482ea094c5a397e791f79e985 to your computer and use it in GitHub Desktop.
Use LightGBM to model the normalized activations of the Autoencoder
import lightgbm as lgb
params = {
'max_depth': 7,
'eta': 0.025,
'objective': 'multiclass',
'num_class': 5,
'metric' : 'multi_logloss',
'num_leaves' : 45,
'bagging_fraction' : 0.5,
'lambda_l2': 200,
'max_bin': 100000,
'feature_fraction':0.05,
}
model = lgb.train(
params,
dtrain,
num_boost_round=10000,
early_stopping_rounds=200,
valid_sets=[dtrain, dval])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment