From 37a606df0a74a66d397db86132917392bcb6bacf Mon Sep 17 00:00:00 2001 From: yzlin Date: Thu, 1 Feb 2024 00:18:53 +0800 Subject: [PATCH] rm unfinished --- kaggle_team.py | 41 ----------------------------------------- 1 file changed, 41 deletions(-) delete mode 100644 kaggle_team.py diff --git a/kaggle_team.py b/kaggle_team.py deleted file mode 100644 index e9f3e67de..000000000 --- a/kaggle_team.py +++ /dev/null @@ -1,41 +0,0 @@ -#!/usr/bin/env python -# -*- coding: utf-8 -*- - -import fire - -from metagpt.roles.kaggle_manager import KaggleManager -from metagpt.roles.ml_engineer import MLEngineer -from metagpt.team import Team - - -async def main( - # competition: str, - # data_desc: str, - # requirement: str, - investment: float = 5.0, - n_round: int = 10, - auto_run: bool = False, -): - competition, data_desc, requirement = ( - "titanic", - "Training set is train.csv.\nTest set is test.csv. We also include gender_submission.csv, a set of predictions that assume all and only female passengers survive, as an example of what a submission file should look like.", - # "Run EDA on the train dataset, train a model to predict survival (20% as validation) and save it, predict the test set using saved model, save the test result according to format", - # "generate a random prediction, replace the Survived column of gender_submission.csv, and save the prediction to a new submission file", - "Score as high as possible for the provided dataset, save the test prediction to a csv with two columns PassengerId and Survived", - ) - - team = Team() - team.hire( - [ - KaggleManager(competition=competition, data_desc=data_desc), - MLEngineer(goal=requirement, auto_run=auto_run), - ] - ) - - team.invest(investment) - team.start_project(requirement) - await team.run(n_round=n_round) - - -if __name__ == "__main__": - fire.Fire(main)