import argparse
import json
# from datetime import datetime, timezone
from pathlib import Path
import yaml
def cargar_config(config_path: str) -> dict:
with open(config_path, encoding="utf-8") as f:
return yaml.safe_load(f)
def generar_html(metricas: dict) -> str:
timestamp_display = metricas.get("timestamp", "N/A")
accuracy = metricas.get("accuracy", 0)
f1 = metricas.get("f1_macro", 0)
roc_auc = metricas.get("roc_auc_ovr", 0)
modelo = metricas.get("model_name", "N/A")
n_estimators = metricas.get("n_estimators", "N/A")
n_test = metricas.get("n_test", "N/A")
if accuracy >= 0.95:
color_acc = "#28a745"
label_acc = "Excelente"
elif accuracy >= 0.85:
color_acc = "#ffc107"
label_acc = "Bueno"
else:
color_acc = "#dc3545"
label_acc = "Revisar"
html = f"""
Reporte de Métricas — Proyecto DS
{accuracy:.4f}
Accuracy
{label_acc}
{roc_auc:.4f}
ROC-AUC (OvR)
Detalles del experimento
| Modelo | {modelo} |
| N° estimadores | {n_estimators} |
| Muestras de test | {n_test} |
| Timestamp (UTC) | {timestamp_display} |
"""
return html
def main(config_path: str):
config = cargar_config(config_path)
metrics_path = config["evaluation"]["output_path"]
print(f"Leyendo métricas desde: {metrics_path}")
with open(metrics_path, encoding="utf-8") as f:
metricas = json.load(f)
html = generar_html(metricas)
report_path = config["report"]["output_path"]
Path(report_path).parent.mkdir(exist_ok=True)
with open(report_path, "w", encoding="utf-8") as f:
f.write(html)
print(f"Reporte guardado en: {report_path}")
pages_path = config["report"]["pages_path"]
Path(pages_path).parent.mkdir(exist_ok=True)
with open(pages_path, "w", encoding="utf-8") as f:
f.write(html)
print(f"Reporte copiado a Pages: {pages_path}")
if __name__ == "__main__":
parser = argparse.ArgumentParser(description="Generador de reporte HTML")
parser.add_argument("--config", default="config/config.yaml")
args = parser.parse_args()
main(args.config)