diff --git a/__pycache__/__init__.cpython-36.pyc b/__pycache__/__init__.cpython-36.pyc new file mode 100644 index 0000000..dddb71a 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..2191ddd 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..61e5b41 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 19f8726..5fe77e5 100644 --- a/q01_bagging/build.py +++ b/q01_bagging/build.py @@ -1,3 +1,4 @@ +# %load q01_bagging/build.py import pandas as pd from sklearn.model_selection import train_test_split from sklearn.tree import DecisionTreeClassifier @@ -15,5 +16,32 @@ # Write your code here +def bagging(X_train, X_test, y_train, y_test, n_est ): + + test_score = [] + train_score = [] + n_esti = range(1,n_est) + for i in n_esti: + bagging_clf2 = BaggingClassifier(DecisionTreeClassifier(), n_estimators=i, max_samples=0.67, + bootstrap=True, random_state=9 ,max_features = 0.67,) + + bagging_clf2.fit(X_train, y_train) + y_pred_bagging = bagging_clf2.predict(X_test) + test_score.append( accuracy_score(y_test, y_pred_bagging)) + + y_pred_bagging1 = bagging_clf2.predict(X_train) + train_score.append( accuracy_score(y_train, y_pred_bagging1)) + + plt.plot(n_esti, test_score , label = 'Test_set') + plt.plot(n_esti,train_score, label = 'Train_set') + plt.xlabel('n_estimators') + + plt.ylabel('accuracy') + plt.legend(loc=1) + 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..7b3a987 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..2685958 Binary files /dev/null and b/q01_bagging/tests/__pycache__/test_q01_bagging.cpython-36.pyc differ