Skip to content

Commit

Permalink
tmp
Browse files Browse the repository at this point in the history
  • Loading branch information
EdenWuyifan committed Aug 14, 2023
1 parent 42a5fb1 commit 801c40a
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion alpha_automl/automl_api.py
Original file line number Diff line number Diff line change
Expand Up @@ -133,8 +133,9 @@ def fit(self, X, y):
leaderboard_data.append([index, pipeline.get_summary(), pipeline.get_score()])

self.leaderboard = pd.DataFrame(leaderboard_data, columns=['ranking', 'pipeline', self.metric])

best_pipeline_id = PIPELINE_PREFIX + '1'
logger.critical(f'[yfw] {self.pipelines[best_pipeline_id].get_pipeline()}')
self._fit(X, y, best_pipeline_id)

def predict(self, X):
Expand Down

0 comments on commit 801c40a

Please sign in to comment.