NLPV commited on
Commit
2d520be
·
verified ·
1 Parent(s): f8dd345

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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=25, 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,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=25, 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
 
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