diff --git a/baselines/utils.py b/baselines/utils.py new file mode 100644 index 0000000..38e5dec --- /dev/null +++ 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