Skip to content

Instantly share code, notes, and snippets.

@ogyalcin

ogyalcin/model_training.py

Created August 2, 2018 12:10
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save ogyalcin/d7ad1be71460b56ebb64bcf8bc369d3a to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save ogyalcin/d7ad1be71460b56ebb64bcf8bc369d3a to your computer and use it in GitHub Desktop.
Download ZIP
Select and Train Model and Make Predictions
Raw
model_training.py
from sklearn.ensemble import GradientBoostingClassifier
model = GradientBoostingClassifier(learning_rate=0.1,max_depth=3)
model.fit(X_train, y_train)
predictions = model.predict(X_test)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment