diff --git a/__pycache__/__init__.cpython-36.pyc b/__pycache__/__init__.cpython-36.pyc new file mode 100644 index 0000000..949493d 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..2df6a3f 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..a5ebe08 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..8f0dcba 100644 --- a/q01_bagging/build.py +++ b/q01_bagging/build.py @@ -1,4 +1,6 @@ +# %load q01_bagging/build.py import pandas as pd +import numpy as np from sklearn.model_selection import train_test_split from sklearn.tree import DecisionTreeClassifier from sklearn.ensemble import BaggingClassifier @@ -12,8 +14,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) - +estimator = np.arange(1, 51) # Write your code here +def bagging(X_train, X_test, y_train, y_test, estimator): + train_accuracy = [] + test_accuracy = [] + + for est in estimator: + bc = BaggingClassifier(base_estimator=DecisionTreeClassifier(), n_estimators=est, + max_samples=0.67, bootstrap=True, random_state=9, + max_features=0.67) + model = bc.fit(X_train, y_train) + + y_train_pred = model.predict(X_train) + train_accuracy.append(accuracy_score(y_train, y_train_pred)) + + y_test_pred = model.predict(X_test) + test_accuracy.append(accuracy_score(y_test, y_test_pred)) + + plt.plot(estimator, train_accuracy) + plt.plot(estimator, test_accuracy) + plt.show() + +bagging(X_train, X_test, y_train, y_test, estimator) 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..2617c07 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..adb279a Binary files /dev/null and b/q01_bagging/tests/__pycache__/test_q01_bagging.cpython-36.pyc differ