File size: 870 Bytes
3a55f64 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 | """
K2 — Domain breadth per-cell (R8 fix).
K2[cell] = distinct domains seen / total domains seen.
"""
from collections import defaultdict
def compute_k2_percell(cells, cell_domains):
"""
Args:
cells: iterable of Cell
cell_domains: {cell_id: set(domain_names)}
Returns:
{cell_id: K2 in [0, 1]}
"""
all_domains = set()
for doms in cell_domains.values():
all_domains.update(doms)
n_total = max(1, len(all_domains))
out = {}
for c in cells:
doms = cell_domains.get(c.cell_id, set())
out[c.cell_id] = len(doms) / n_total
return out
def update_cell_domains(cell_domains, cells, active_ids, domain_name):
"""Add domain_name to each active cell's domain set."""
for cid in active_ids:
cell_domains.setdefault(cid, set()).add(domain_name) |