Skip to content

Commit

Permalink
adding dataset type as column to the result collection
Browse files Browse the repository at this point in the history
  • Loading branch information
danielbinschmid committed Sep 14, 2024
1 parent 6c99158 commit 58d8bf2
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions code/experiments/utils/result_collection.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@ def __init__(
) -> None:
self.data = data
self.config = config
self.ds_type = config.ds_type


class ResultCollection:
Expand Down Expand Up @@ -38,6 +39,7 @@ def save(self, t_file_prefix: str = "res"):
result = x.data
result["type_model"] = x.config.conf_model.type.name.lower()
result["transform"] = x.config.transforms.name.lower()
result["ds_type"] = x.config.ds_type.name.lower()
data.append(x.data)

df = pd.DataFrame(data)
Expand Down

0 comments on commit 58d8bf2

Please sign in to comment.