Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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(
|
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,
|
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))
|