def fit(self, path=None):
"""Fit transformers"""
path = path if path else self.path
t0 = time()
xtemp, ytemp = slice_array(
self.in_array, self.targets, self.in_index)
t0_f = time()
self.estimator.fit(xtemp, ytemp)
self.transform_time_ = time() - t0_f
if self.out_array is not None:
self._transform()
o = IndexedEstimator(estimator=self.estimator,
name=self.name_index,
index=self.index,
in_index=self.in_index,
out_index=self.out_index,
data=self.data)
save(path, self.name_index, o)
if self.verbose:
f = "stdout" if self.verbose < 10 else "stderr"
msg = "{:<30} {}".format(self.name_index, "done")
print_time(t0, msg, file=f)
评论列表
文章目录