fix: update metric direction to return bool (#791)

This commit is contained in:
you-n-g
2025-04-15 17:10:31 +08:00
committed by GitHub
parent ea5fef308a
commit b18ed7ce27
2 changed files with 5 additions and 2 deletions
+1
View File
@@ -200,6 +200,7 @@ def main(
DS_RD_SETTING.competition = competition
if DS_RD_SETTING.competition:
if DS_RD_SETTING.scen.endswith("KaggleScen"):
download_data(competition=DS_RD_SETTING.competition, settings=DS_RD_SETTING)
else:
@@ -31,7 +31,9 @@ class DataScienceScen(Scenario):
self.raw_description = self._get_description()
self.processed_data_folder_description = self._get_data_folder_description()
self._analysis_competition_description()
self.metric_direction = self._get_direction()
self.metric_direction: bool = (
self._get_direction()
) # True indicates higher is better, False indicates lower is better
def _get_description(self):
if (fp := Path(f"{DS_RD_SETTING.local_data_path}/{self.competition}.json")).exists():
@@ -148,7 +150,7 @@ class KaggleScen(DataScienceScen):
def _get_direction(self):
leaderboard = leaderboard_scores(self.competition)
return "maximize" if float(leaderboard[0]) > float(leaderboard[-1]) else "minimize"
return float(leaderboard[0]) > float(leaderboard[-1])
@property
def rich_style_description(self) -> str: