diff --git a/__pycache__/__init__.cpython-36.pyc b/__pycache__/__init__.cpython-36.pyc new file mode 100644 index 0000000..6d4e3f4 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..7c7022d 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..063f993 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..ca231a2 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 @@ -5,6 +6,7 @@ import matplotlib.pyplot as plt from sklearn.metrics import accuracy_score plt.switch_backend('agg') +import numpy as np # Data Loading dataframe = pd.read_csv('data/loan_prediction.csv') @@ -12,8 +14,36 @@ 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) - +n_est = 1 # Write your code here +def bagging(X_train, X_test, y_train, y_test,n_est): + + i=1 + train = dict() + test = dict() + + while (i<=50): + + bagging_clf = BaggingClassifier(DecisionTreeClassifier(), n_estimators=i, max_samples=0.67, + bootstrap=True, random_state=9,max_features=0.67) + model = bagging_clf.fit(X_train, y_train) + + y_pred_train = model.predict(X_train) + score_train = accuracy_score(y_train, y_pred_train) + + y_pred_test = model.predict(X_test) + score_test = accuracy_score(y_test, y_pred_test) + + train[i]=score_train + test[i]=score_test + i= i + 1 + + plt.plot(np.arange(1,51),train.values()) + plt.plot(np.arange(1,51),test.values()) + plt.show() + +# bagging(X_train, X_test, y_train, y_test,n_est) + 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..305a5f8 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..7421c12 Binary files /dev/null and b/q01_bagging/tests/__pycache__/test_q01_bagging.cpython-36.pyc differ