From 18bd11c09fdc9ab53018670b42817fa4398e4268 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Daniel=20Alves=20R=C3=B6sel?=
 <60182044+velocitatem@users.noreply.github.com>
Date: Fri, 12 Dec 2025 12:55:52 +0100
Subject: [PATCH] Update experiments/ml/train.py

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
---
 experiments/ml/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/experiments/ml/train.py b/experiments/ml/train.py
index 8766114..10e4fe8 100644
--- a/experiments/ml/train.py
+++ b/experiments/ml/train.py
@@ -83,7 +83,7 @@ def train(data_path=None, model_type='xgboost', test_size=0.2, random_state=42,
     # class distribution
     n_agents = y.sum()
     n_humans = (y == 0).sum()
-    logger.info(f"Class distribution: {n_humans} humans, {n_agents} agents (ratio {n_humans/n_agents:.2f})")
+    logger.info(f"Class distribution: {n_humans} humans, {n_agents} agents" + (f" (ratio {n_humans / n_agents:.2f})" if n_agents > 0 else ""))
 
     # train/test split with stratification
     X_train, X_test, y_train, y_test = train_test_split(