diff --git a/__pycache__/__init__.cpython-36.pyc b/__pycache__/__init__.cpython-36.pyc new file mode 100644 index 0000000..8d877cc 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..d5e983a 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..de3970c 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..8186ae4 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 @@ -12,8 +13,29 @@ X = dataframe.iloc[:, :-1] y = dataframe.iloc[:, -1] X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=9) +def bagging(X_train, X_test, y_train, y_test, n_est): + acc_train, acc_test = [], [] + ranges = range(1, n_est, 2) + for n_est in ranges: + clf_bagging_20 = BaggingClassifier(DecisionTreeClassifier(), n_estimators=n_est, max_samples=0.67, + max_features=0.67, bootstrap=True, random_state=9) + clf_bagging_20.fit(X_train, y_train) + X_pred_bagging = clf_bagging_20.predict(X_train) + y_pred_bagging = clf_bagging_20.predict(X_test) + acc_score_train = accuracy_score(y_train, X_pred_bagging) + acc_score_test = accuracy_score(y_test, y_pred_bagging) + acc_train.append(acc_score_train) + acc_test.append(acc_score_test) + plt.figure(figsize=(10, 6)) + plt.plot(ranges, acc_train, c='b', label='Train set') + plt.plot(ranges, acc_test, c='g', label='Test set') + plt.legend(loc='upper right') + plt.xlabel('n_estimators') + plt.ylabel('Accuracy') + plt.show() +# Write your code here + -# Write your code here 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..b7ea2a6 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..0347cef Binary files /dev/null and b/q01_bagging/tests/__pycache__/test_q01_bagging.cpython-36.pyc differ