Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -24,7 +24,7 @@ def run_dare_demo(delete_index=25):
|
|
24 |
logs = ""
|
25 |
|
26 |
# Train initial model
|
27 |
-
model = dare.Forest(n_estimators=
|
28 |
start = time.perf_counter()
|
29 |
model.fit(X_train, y_train)
|
30 |
train_time = time.perf_counter() - start
|
@@ -57,7 +57,7 @@ def run_dare_demo(delete_index=25):
|
|
57 |
X_retrain = np.delete(X_train, delete_index, axis=0)
|
58 |
y_retrain = np.delete(y_train, delete_index, axis=0)
|
59 |
|
60 |
-
retrain_model = dare.Forest(n_estimators=
|
61 |
start_retrain = time.perf_counter()
|
62 |
retrain_model.fit(X_retrain, y_retrain)
|
63 |
retrain_time = time.perf_counter() - start_retrain
|
|
|
24 |
logs = ""
|
25 |
|
26 |
# Train initial model
|
27 |
+
model = dare.Forest(n_estimators=50, max_depth=3, random_state=42)
|
28 |
start = time.perf_counter()
|
29 |
model.fit(X_train, y_train)
|
30 |
train_time = time.perf_counter() - start
|
|
|
57 |
X_retrain = np.delete(X_train, delete_index, axis=0)
|
58 |
y_retrain = np.delete(y_train, delete_index, axis=0)
|
59 |
|
60 |
+
retrain_model = dare.Forest(n_estimators=50, max_depth=3, random_state=42)
|
61 |
start_retrain = time.perf_counter()
|
62 |
retrain_model.fit(X_retrain, y_retrain)
|
63 |
retrain_time = time.perf_counter() - start_retrain
|