vrr/WFGY
2
0
Fork 0
mirror of https://github.com/onestardao/WFGY.git synced 2026-05-20 09:23:03 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

Update example_02_self_reflection.py

Browse source
This commit is contained in:
PSBigBig 2025-06-11 21:13:14 +08:00 • committed by GitHub
parent 9168f2b5f8
commit bb59cd36e8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 14 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

20
examples/example_02_self_reflection.py
View file

@ -1,17 +1,25 @@
# example_02_self_reflection.py
# Three successive runs on a single engine
# Three successive runs on one engine with metrics
import pathlib, sys
import pathlib, sys, numpy as np
sys.path.insert(0, str(pathlib.Path(__file__).resolve().parents[1]))
import numpy as np, wfgy_sdk as w
import wfgy_sdk as w
from wfgy_sdk.evaluator import compare_logits
rng = np.random.default_rng(1)
eng = w.get_engine(reload=True)
print("\n=== Example 02 · Self-reflection loop ===")
for step in range(3):
G = rng.normal(size=64); G /= np.linalg.norm(G)
I = G + rng.normal(scale=0.05, size=64)
logits = rng.normal(size=4096)
logits_before = rng.normal(size=4096)
out = eng.run(input_vec=I, ground_vec=G, logits=logits)
print(f"[Round {step}] logit[0] after WFGY = {out[0]:.4f}")
logits_after = eng.run(input_vec=I, ground_vec=G, logits=logits_before)
m = compare_logits(logits_before, logits_after)
print(f"[Round {step}] "
f"variance ratio {m['std_ratio']:.2f} | "
f"KL {m['kl_divergence']:.2f} | "
f"top-1 shift {'✔' if m['top1_shift'] else '✘'}")