feat: enhance training pipeline with research templates and experiment configuration

This commit is contained in:
2025-08-08 23:48:55 +02:00
parent 96291b4ad0
commit 6d39c3afc1
9 changed files with 341 additions and 755 deletions
+6 -1
View File
@@ -33,6 +33,7 @@ class ModelTrainer:
model_type: str = "logistic_regression",
features: List[str] = None,
model_params: Dict[str, Any] = None,
tags: List[str] = None,
save_artifacts: bool = True,
) -> str:
"""
@@ -45,6 +46,10 @@ class ModelTrainer:
features = ["full_name"]
feature_types = [FeatureType(f) for f in features]
# Prepare tags - combine default tags with template tags
default_tags = ["training", model_type]
experiment_tags = default_tags + (tags or [])
# Create experiment configuration
config = ExperimentConfig(
name=model_name,
@@ -52,7 +57,7 @@ class ModelTrainer:
model_type=model_type,
features=feature_types,
model_params=model_params or {},
tags=["training", model_type],
tags=experiment_tags,
)
# Run experiment