Skip to content

Commit

Permalink
out of bag evaluation
Browse files Browse the repository at this point in the history
  • Loading branch information
LordSomen committed Jul 24, 2018
1 parent e64e0b3 commit 5bb4511
Showing 1 changed file with 12 additions and 0 deletions.
12 changes: 12 additions & 0 deletions Ensemble_learning/ensemble_learning.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,8 @@
print(Y_train)

#%%
''' voting classifier'''

from sklearn.ensemble import RandomForestClassifier
from sklearn.linear_model import LogisticRegression
from sklearn.svm import SVC
Expand All @@ -36,6 +38,7 @@
print(clf.__class__.__name__, accuracy_score(Y_test, Y_pred))

#%%
''' bagging and pasting'''
from sklearn.tree import DecisionTreeClassifier
from sklearn.ensemble import BaggingClassifier

Expand All @@ -49,3 +52,12 @@
prediction = bag_clf.predict(X_test)
print(accuracy_score(Y_test,prediction))

#%%
'''out of bag evaluation'''
bag_clf = BaggingClassifier(DecisionTreeClassifier(),
n_estimators=500,
max_samples=100,bootstrap=True,n_jobs=-1,
oob_score=True)
bag_clf.fit(X_train,Y_train)
print(bag_clf.oob_score_)

0 comments on commit 5bb4511

Please sign in to comment.