tune n_estimators --> v0.6.0 Pt.2

thierrymoudiki · Aug 9, 2024 · 63817a2 · 63817a2
1 parent 5cf1658
commit 63817a2
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/examples/lazy_cross_val.py b/examples/lazy_cross_val.py
@@ -56,3 +56,25 @@
 print(f"Elapsed: {time()-start}")
 #print(res3)
 print(res4)
+
+
+start = time()
+res4 = ub.lazy_cross_val_optim(
+    X_train,
+    y_train,
+    X_test=X_test,
+    y_test=y_test,
+    model_type="lightgbm",
+    type_fit="classification",
+    scoring="accuracy",
+    cv=5,
+    n_jobs=None,
+    n_init=10,
+    n_iter=190,
+    abs_tol=1e-3,
+    seed=123,
+    customize=False
+)
+print(f"Elapsed: {time()-start}")
+#print(res3)
+print(res4)