Skip to content

Commit 699f62a

Browse files
committed
RandomizedSearchCV parameter definition and initialization into separate Marimo cells and update generated version.
1 parent 729ecd9 commit 699f62a

1 file changed

Lines changed: 6 additions & 13 deletions

File tree

research.py

Lines changed: 6 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import marimo
22

3-
__generated_with = "0.17.8"
3+
__generated_with = "0.18.4"
44
app = marimo.App()
55

66

@@ -239,18 +239,7 @@ def _(mo):
239239

240240

241241
@app.cell
242-
def _(
243-
RandomizedSearchCV,
244-
lda,
245-
lr_model,
246-
np,
247-
pca,
248-
pipe,
249-
rf_model,
250-
x_train,
251-
xgb_model,
252-
y_train,
253-
):
242+
def _(lda, lr_model, np, pca, rf_model, xgb_model):
254243
param_list = [
255244
{ # Random Forest, PCA On
256245
"model": [rf_model],"model__n_estimators":np.arange(150,650,100),
@@ -279,7 +268,11 @@ def _(
279268
"model": [xgb_model], "model__n_estimators" : [500,700,900],"model__learning_rate": [0.01,0.1], "model__max_depth":np.arange(7,14,3)
280269
}
281270
]
271+
return (param_list,)
282272

273+
274+
@app.cell
275+
def _(RandomizedSearchCV, param_list, pipe, x_train, y_train):
283276
rscv = RandomizedSearchCV(
284277
estimator=pipe,param_distributions=param_list,n_iter=8,cv=5,n_jobs=-1,random_state=50,refit=True
285278
)

0 commit comments

Comments
 (0)