new baseline utils added

2023-11-22 19:18:52 +01:00 · 2023-11-22 19:18:52 +01:00 · 907650ea7e
parent 91cdb87323
commit 907650ea7e
1 changed files with 8 additions and 0 deletions
--- a/baselines/utils.py
+++ b/baselines/utils.py
@ -0,0 +1,8 @@
+from sklearn import clone
+from sklearn.base import BaseEstimator
+
+
+def clone_fit(c_model: BaseEstimator, data, labels):
+    c_model2 = clone(c_model)
+    c_model2.fit(data, labels)
+    return c_model2