PreyPatel commited on
Commit
384323b
1 Parent(s): 54c6fe2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -89,7 +89,7 @@ if ensemble_type == "bagging":
89
  test_loss = []
90
  train_loss = []
91
  for i in range(1, estimator_number):
92
- model = BaggingRegressor(estimator=estimator, n_estimators=i, random_state=45)
93
  model.fit(X_train, y_train)
94
  y_pred = model.predict(X_test)
95
  temp = mean_squared_error(y_test, y_pred)
@@ -112,7 +112,7 @@ elif ensemble_type == "boosting":
112
  test_loss = []
113
  estimator = estimator_model(model_type)
114
  for i in range(1, estimator_number):
115
- model = AdaBoostRegressor(n_estimators=i, estimator=estimator)
116
  model.fit(X_train, y_train)
117
  y_pred = model.predict(X_test)
118
  test_loss.append(mean_squared_error(y_test, y_pred))
 
89
  test_loss = []
90
  train_loss = []
91
  for i in range(1, estimator_number):
92
+ model = BaggingRegressor(base_estimator=estimator, n_estimators=i, random_state=45)
93
  model.fit(X_train, y_train)
94
  y_pred = model.predict(X_test)
95
  temp = mean_squared_error(y_test, y_pred)
 
112
  test_loss = []
113
  estimator = estimator_model(model_type)
114
  for i in range(1, estimator_number):
115
+ model = AdaBoostRegressor(n_estimators=i, base_estimator=estimator)
116
  model.fit(X_train, y_train)
117
  y_pred = model.predict(X_test)
118
  test_loss.append(mean_squared_error(y_test, y_pred))