fix: add github workflow

This commit is contained in:
2025-10-07 23:21:35 +02:00
parent d3b3840278
commit f2ac0c9769
25 changed files with 307 additions and 89 deletions
+13 -4
View File
@@ -30,12 +30,21 @@ def train_from_template(
logging.info(f"Features: {experiment_config.get('features')}")
trainer = ModelTrainer(cfg)
name_val = experiment_config.get("name")
type_val = experiment_config.get("model_type")
features_val = experiment_config.get("features") or ["full_name"]
tags_val = experiment_config.get("tags", [])
if not isinstance(name_val, str) or not isinstance(type_val, str):
raise ValueError("Template must include 'name' and 'model_type' as strings")
if not isinstance(features_val, list):
raise ValueError("Template 'features' must be a list of strings")
trainer.train_single_model(
model_name=experiment_config.get("name"),
model_type=experiment_config.get("model_type"),
features=experiment_config.get("features"),
model_name=name_val,
model_type=type_val,
features=features_val,
model_params=experiment_config.get("model_params", {}),
tags=experiment_config.get("tags", []),
tags=tags_val if isinstance(tags_val, list) else [],
)
logging.info("Training completed successfully!")