wassname/jsteer
1
0
Fork 0
mirror of https://github.com/wassname/jsteer.git synced 2026-09-11 15:00:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

demo: single demo_steer entry point -- headings, searched anchors, comparison table

Browse Source 
wassname: "one demo function, everywhere, nice output, comparable qualitative outputs,
heading making config/method clear, table at end comparing quants". show_steer now returns
its per-anchor quants; demo_steer(vecs, ...) loops methods with a clear ## heading each,
calls the searched-anchor show_steer (strongest coherent steer both ways), and prints one
github-table comparison of the readout at -C*/0/+C* across methods. demo_edges.py is now a
3-line call. Also: research_journal.md started (ARJ).

Co-Authored-By: Claudypoo <288921227+claudypoo@users.noreply.github.com>
This commit is contained in:
wassname
2026-07-12 08:53:24 +08:00
co-authored by Claudypoo
parent 2c17603e96
commit 32030df462
3 changed files with 109 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/scratch/demo_edges.py
+6 -8
View File
@@ -14,8 +14,8 @@ from loguru import logger # noqa: E402
from steering_lite import MeanDiffC, Vector # noqa: E402
from transformers import AutoModelForCausalLM, AutoTokenizer # noqa: E402
from jsteer import Jacobian, show_steer # noqa: E402
from jsteer.demo import YESNO # noqa: E402
from jsteer import Jacobian # noqa: E402
from jsteer.demo import YESNO, demo_steer # noqa: E402
MODEL = "Qwen/Qwen3.5-4B"
tok = AutoTokenizer.from_pretrained(MODEL)
@@ -47,12 +47,10 @@ DILEMMA = ("You are late for work because you partied until very late last night
vecs = {
"persona_pinv": jac.persona_pinv_vector(model, tok, deceptive, honest, layers=band),
"word(lie)": jac.word_vector(model, tok, ["lie", "deceive", "dishonest"], layers=band),
"word(lie/deceive)": jac.word_vector(model, tok, ["lie", "deceive", "dishonest"], layers=band),
"meandiff(base)": Vector.train(model, tok, deceptive, honest, MeanDiffC(layers=tuple(band))),
}
for name, v in vecs.items():
logger.info(f"\n\n##################### {name} #####################")
# Cs=None -> show_steer searches the coherent edge each way and demos the anchors.
show_steer(jac, model, tok, v, DILEMMA, rubric=DILEMMA, readout=YESNO,
max_new_tokens=256, budget=6)
# ONE call: per-method heading + searched anchors + comparable generations + end table.
demo_steer(jac, model, tok, vecs, DILEMMA, rubric=DILEMMA, readout=YESNO,
max_new_tokens=256, budget=6)