Views
No views yet
SallySims/equibert-generator| Prefix | Task | Input → Output |
|---|---|---|
rewrite inclusive: | Inclusive rewriting | Biased text → Inclusive version |
generate policy: | Policy generation | Topic → Policy section |
generate jd: | Job description | Role description → Inclusive JD |
rewrite framing: | Framing correction | Victim-blaming text → Structural framing |
generate commitment: | DEI commitment | Goal → Measurable commitment |
rewrite review: | Review debiasing | Biased review → Unbiased version |
generate awareness: | Awareness content | Topic → Awareness statement |
1from transformers import T5ForConditionalGeneration, T5Tokenizer
2
3model = T5ForConditionalGeneration.from_pretrained("SallySims/equibert-generator")
4tokenizer = T5Tokenizer.from_pretrained("SallySims/equibert-generator")
5
6prompt = "rewrite inclusive: We need a rock star developer who can dominate the roadmap."
7inputs = tokenizer(prompt, return_tensors="pt", max_length=256, truncation=True)
8output = model.generate(**inputs, max_new_tokens=200, num_beams=4)
9result = tokenizer.decode(output[0], skip_special_tokens=True)
10print(result)
11# "We are looking for a skilled developer with strong technical expertise
12# who can contribute meaningfully to our product roadmap."Input Text
│
├──▶ RoBERTa-base encoder ──▶ Linear projection
│ │
└──▶ DeBERTa-v3-base encoder ──▶ Linear projection
│
Weighted fusion (learned α)
│
Layer Norm + Dropout
│
Task-specific head (see below)1@misc{equibert2024,
2 author = {SallySims},
3 title = {EquiBERT: A Multi-Task DEI Transformer},
4 year = {2024},
5 publisher = {HuggingFace},
6 url = {https://huggingface.co/SallySims}
7}| Model | Task | Primary Metric |
|---|---|---|
| equibert-bias-classifier | Bias Detection | Macro F1 |
| equibert-microaggression | Microaggression Detection | Macro F1 |
| equibert-category-tagger | DEI Category Tagging | Macro F1 |
| equibert-event-exclusion | Event Exclusion Classification | Macro F1 |
| equibert-inclusive-language | Inclusive Language Scoring | Span F1 |
| equibert-review-auditor | Performance Review Auditing | Span F1 |
| equibert-washing-detector | DEI Washing Detection | MAE |
| equibert-framing-scorer | Report Framing Scoring | MAE |
| equibert-awareness-scorer | DEI Awareness Scoring | MAE |
| equibert-similarity | Semantic Similarity | Accuracy |
| equibert-ner | DEI Entity Recognition | Span F1 |
| equibert-relation-extraction | Relation Extraction | Macro F1 |
| equibert-qa | Extractive QA | Span EM |
| equibert-search | Semantic Search | MRR@10 |
| equibert-nli | NLI / Textual Entailment | Macro F1 |
| equibert-generator | DEI Text Generation | ROUGE-L |