Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- from sklearn.ensemble import GradientBoostingClassifier
- X_train, X_test, y_train, y_test = train_test_split(X, y, random_state = 0)
- clf = GradientBoostingClassifier(random_state = 0)
- clf.fit(X_train, y_train)
- print('Breast cancer dataset (learning_rate=0.1, max_depth=3)')
- print('Accuracy of GBDT classifier on training set: {:.2f}'
- .format(clf.score(X_train, y_train)))
- print('Accuracy of GBDT classifier on test set: {:.2f}\n'
- .format(clf.score(X_test, y_test)))
- clf = GradientBoostingClassifier(learning_rate = 0.01, max_depth = 2, random_state = 0)
- clf.fit(X_train, y_train)
- print('Breast cancer dataset (learning_rate=0.01, max_depth=2)')
- print('Accuracy of GBDT classifier on training set: {:.2f}'
- .format(clf.score(X_train, y_train)))
- print('Accuracy of GBDT classifier on test set: {:.2f}'
- .format(clf.score(X_test, y_test)))
Add Comment
Please, Sign In to add comment