Spaces:
Sleeping
Sleeping
| from __future__ import annotations | |
| import json | |
| from pathlib import Path | |
| import pandas as pd | |
| def _format_evidence_item(item: dict[str, Any]) -> str: | |
| if "top_B_values" in item: | |
| top_vals = ", ".join( | |
| f"{x['value']}={x['probability']:.2f}" | |
| for x in item.get("top_B_values", []) | |
| ) | |
| return f"- `{item.get('A_value', '')}` (n={item.get('count', 0)}): {top_vals}" | |
| if "B_value" in item: | |
| match_type = item.get("match_type", "match") | |
| return f"- `{item.get('A_value', '')}` -> `{item.get('B_value', '')}` ({match_type})" | |
| return f"- `{item}`" | |
| def write_markdown_report(df: pd.DataFrame, path: str | Path, top_n: int = 15) -> None: | |
| path = Path(path) | |
| path.parent.mkdir(parents=True, exist_ok=True) | |
| lines = ["# Strongest property relationships", ""] | |
| if df.empty: | |
| lines.append("No relationships were detected.") | |
| path.write_text("\n".join(lines), encoding="utf-8") | |
| return | |
| for i, row in df.head(top_n).iterrows(): | |
| lines.append(f"## {i + 1}. {row['A']} → {row['B']}") | |
| lines.append(f"- classification: **{row['classification']}**") | |
| lines.append(f"- strength: `{row['strength']:.3f}`") | |
| lines.append(f"- predictive strength: `{row['predictive_strength']:.3f}`") | |
| lines.append(f"- support: `{row['support']:.3f}`") | |
| lines.append(f"- determinism: `{row['determinism']:.3f}`") | |
| lines.append(f"- stability: `{row['stability']:.3f}`") | |
| lines.append(f"- doc alignment: `{row['doc_alignment']:.3f}`") | |
| lines.append(f"- heldout accuracy: `{row['heldout_accuracy']:.3f}`") | |
| lines.append(f"- baseline accuracy: `{row['baseline_accuracy']:.3f}`") | |
| lines.append("") | |
| # evidence = row.get("evidence", []) or [] | |
| # if evidence: | |
| # lines.append("Evidence:") | |
| # for item in evidence[:3]: | |
| # lines.append(f"- `{item['A_value']}` (n={item.get('count', 0)}): " + ", ".join(f"{x['value']}={x['probability']:.2f}" for x in item["top_B_values"])) | |
| # lines.append("") | |
| evidence = row.get("evidence", []) | |
| if isinstance(evidence, str): | |
| try: | |
| evidence = json.loads(evidence) | |
| except Exception: | |
| evidence = [] | |
| if evidence: | |
| lines.append("Evidence:") | |
| for item in evidence: | |
| if isinstance(item, dict): | |
| lines.append(_format_evidence_item(item)) | |
| else: | |
| lines.append(f"- `{item}`") | |
| lines.append("") | |
| path.parent.mkdir(parents=True, exist_ok=True) | |
| path.write_text("\n".join(lines), encoding="utf-8") | |
| path.write_text("\n".join(lines), encoding="utf-8") | |