commit-live-students · abkedar · Sep 13, 2018
diff --git a/__pycache__/__init__.cpython-36.pyc b/__pycache__/__init__.cpython-36.pyc
diff --git a/q01_bagging/__pycache__/__init__.cpython-36.pyc b/q01_bagging/__pycache__/__init__.cpython-36.pyc
diff --git a/q01_bagging/__pycache__/build.cpython-36.pyc b/q01_bagging/__pycache__/build.cpython-36.pyc
diff --git a/q01_bagging/build.py b/q01_bagging/build.py
@@ -12,8 +12,31 @@
 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_estimator = range(1, 50)
 
 # Write your code here
 
+def bagging(X_train, X_test, y_train, y_test, n_ests):
+    acc_train, acc_test = [], []
+    ranges = range(1, n_ests, 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')
+
+
+bagging(X_train, X_test, y_train, y_test, n_ests = 50)
+
 
diff --git a/q01_bagging/tests/__pycache__/__init__.cpython-36.pyc b/q01_bagging/tests/__pycache__/__init__.cpython-36.pyc
diff --git a/q01_bagging/tests/__pycache__/test_q01_bagging.cpython-36.pyc b/q01_bagging/tests/__pycache__/test_q01_bagging.cpython-36.pyc