diff --git a/__pycache__/__init__.cpython-36.pyc b/__pycache__/__init__.cpython-36.pyc new file mode 100644 index 0000000..5241cef Binary files /dev/null and b/__pycache__/__init__.cpython-36.pyc differ diff --git a/q01_bagging/__pycache__/__init__.cpython-36.pyc b/q01_bagging/__pycache__/__init__.cpython-36.pyc new file mode 100644 index 0000000..6f8de0b Binary files /dev/null and b/q01_bagging/__pycache__/__init__.cpython-36.pyc differ diff --git a/q01_bagging/__pycache__/build.cpython-36.pyc b/q01_bagging/__pycache__/build.cpython-36.pyc new file mode 100644 index 0000000..689f766 Binary files /dev/null and b/q01_bagging/__pycache__/build.cpython-36.pyc differ diff --git a/q01_bagging/build.py b/q01_bagging/build.py index c34fb02..0819cf0 100644 --- a/q01_bagging/build.py +++ b/q01_bagging/build.py @@ -1,3 +1,4 @@ + import pandas as pd from sklearn.model_selection import train_test_split from sklearn.tree import DecisionTreeClassifier @@ -14,5 +15,28 @@ # Write your code here +def bagging(X_train, X_test, y_train, y_test,n_est): + accuracy_scores=[] + accuracy_scores_t=[] + n_est_list=[] + for n in list(range(1,n_est,1)): + bagging_clf1 = BaggingClassifier(DecisionTreeClassifier(), n_estimators=n, max_samples=0.67,max_features=0.67, + bootstrap=True, random_state=9) + bagging_clf1.fit(X_train, y_train) + y_pred_bagging = bagging_clf1.predict(X_test) + score_bc_lr = accuracy_score(y_test, y_pred_bagging) + accuracy_scores.append(score_bc_lr) + + y_pred_bagging_t = bagging_clf1.predict(X_train) + score_bc_lr_t = accuracy_score(y_train, y_pred_bagging_t) + accuracy_scores_t.append(score_bc_lr_t) + n_est_list.append(n) + + plt.plot(n_est_list,accuracy_scores_t,accuracy_scores) + #plt.legend('Train Set','Test Set') + plt.show() + +#bagging(X_train, X_test, y_train, y_test,50) + diff --git a/q01_bagging/tests/__pycache__/__init__.cpython-36.pyc b/q01_bagging/tests/__pycache__/__init__.cpython-36.pyc new file mode 100644 index 0000000..4a02b1f Binary files /dev/null and b/q01_bagging/tests/__pycache__/__init__.cpython-36.pyc differ diff --git a/q01_bagging/tests/__pycache__/test_q01_bagging.cpython-36.pyc b/q01_bagging/tests/__pycache__/test_q01_bagging.cpython-36.pyc new file mode 100644 index 0000000..55134e4 Binary files /dev/null and b/q01_bagging/tests/__pycache__/test_q01_bagging.cpython-36.pyc differ