def fit(self, x, y=None):
x, y = check_X_y(x, y)
x_train, x_test, y_train, y_test = train_test_split(x, y, random_state=self.random_state)
self.front = run_ffx(x_train, x_test, y_train, y_test,
self.exponents, self.operators, num_alphas=self.num_alphas, l1_ratios=self.l1_ratios,
target_score=self.target_score, n_tail=self.n_tail, random_state=self.random_state,
strategies=self.strategies, n_jobs=self.n_jobs, max_complexity=self.max_complexity,
rational=self.rational, eps=self.eps, **self.kw)
self.make_model(x_test, y_test)
return self
评论列表
文章目录