# Author - Deepak Soni
"""G4 diagnostic — which faculty causes the security forgetting regression? (shared-head, high interference)"""
import os, sys, random
sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))
import torch
from antahkarana.core import Antahkarana
from antahkarana.domains.security import build_security_stream, SecurityBackbone

stream, zd, kn, in_dim = build_security_stream(seed=0)


def run(name, **kw):
    random.seed(0); torch.manual_seed(0)
    bb = SecurityBackbone(in_dim, len(stream), 256, 0)
    m = Antahkarana(bb, epochs=15, bs=128, base_lr=3e-4, avidya_strategy="energy", **kw)
    r = m.run(stream)
    print("%-30s acc=%.4f forgetting=%+.4f" % (name, r["final_avg"], r["forgetting"]), flush=True)


run("naive", samskara=False, replay=False, sleep=False)
run("+samskara", samskara=True, replay=False, sleep=False)
run("+samskara+replay", samskara=True, replay=True, sleep=False)
run("+samskara+replay+sleep (full)", samskara=True, replay=True, sleep=True)