gitea_admin/sientia-dataops-model-manager
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SIENTIAPDE-1252: Remove experiment_description from TrainModelParams and related tests.

Browse Source
This commit is contained in:
Bruno Domingues
2025-10-09 17:48:43 -03:00
parent 5b0cd92dfe
commit 94e11df803
5 changed files with 5 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
model_manager/utils/models/train_model_params.py
View File

@@ -34,7 +34,6 @@ class TrainModelParams:
shuffle (bool): Whether to shuffle the data during train/test split.
experiment_run_id (int): Unique identifier for the experiment run.
experiment_name (str): Name of the experiment for tracking.
experiment_description (str): Description of the experiment.
removed_intervals (list): List of time intervals to remove from the data.
"""
@@ -56,7 +55,6 @@ class TrainModelParams:
shuffle: bool
experiment_run_id: int
experiment_name: str
experiment_description: str
removed_intervals: list
@classmethod
@@ -114,9 +112,6 @@ class TrainModelParams:
data.get('experiment_run_id'), int, 'experiment_run_id'
),
experiment_name=cls._check_none(data.get('experiment_name'), str, 'experiment_name'),
experiment_description=cls._check_none(
data.get('experiment_description'), str, 'experiment_description'
),
removed_intervals=cls._check_type(
data.get('removed_intervals'), list, 'removed_intervals'
),