Skip to content

Instantly share code, notes, and snippets.

@avishekrk
Created August 8, 2016 15:31
Show Gist options
  • Save avishekrk/dc46af4885fb1c44c8d237cb8b3e44da to your computer and use it in GitHub Desktop.
Save avishekrk/dc46af4885fb1c44c8d237cb8b3e44da to your computer and use it in GitHub Desktop.
#YAML file containing the configuration
#of the models.
#DATE STARTS AT THE END OF THE YEAR E.G. DEC 31, 2013
features_table: 'features.combined'
break_window: '1Year'
static_features: ['diameters','pipe_age']
past_cols: ['wo_curr', 'wo_last1', 'wo_last2', 'wo_last3', 'wo_last4', 'wo_last5']
future_cols: ['wo_next1','wo_next2','wo_next3']
visualize: True
debug: True
cross_val:
thirty_seventy: [.3,.7]
models: ['LR','NN','DT', 'RF','NB','ET','SGD','AB','GB']
parameters:
LR:
C: [0.00001, 0.0001, 0.001, 0.01, 0.1]
penalty: ['l1','l2']
#add the NN models here too.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment