Skip to content

feat(datasets): export Aegis violence rows as HumanLabeledDataset - #2524

Open
Ray Winkelman (rwinkelman) wants to merge 3 commits into
microsoft:mainfrom
rwinkelman:feat/aegis-human-labeled-violence-2475
Open

feat(datasets): export Aegis violence rows as HumanLabeledDataset#2524
Ray Winkelman (rwinkelman) wants to merge 3 commits into
microsoft:mainfrom
rwinkelman:feat/aegis-human-labeled-violence-2475

Conversation

@rwinkelman

Copy link
Copy Markdown

Summary

  • Adds fetch_human_labeled_harm_dataset_async() on the Aegis 2.0 remote loader.
  • Retains assistant response + response_label and maps them into HarmHumanLabeledEntry rows for scorer evaluation.
  • Starts with the Violence category (CC-BY-4.0, clean mapping to violence.yaml).

Addresses #2475.

Test plan

  • pytest tests/unit/datasets/test_aegis_human_labeled_dataset.py

Adds fetch_human_labeled_harm_dataset_async so remote Aegis 2.0
response labels can feed scorer evaluation without hand-authored CSVs.

Addresses microsoft#2475 (violence category first).
…icrosoft#2475)

Auto-select harm_definition from harm_category so Hate/Identity Hate rows
build a HumanLabeledDataset against hate_speech.yaml.
@rwinkelman

Copy link
Copy Markdown
Author

Extended with hate_speech profile: HATE_IDENTITY_HATE now auto-selects hate_speech.yaml and builds aegis_hate_speech HumanLabeledDataset.

Map Malware rows to cyber.yaml and add sexual/self_harm harm_definition profiles.
@rwinkelman

Copy link
Copy Markdown
Author

Follow-up: added malware→cyber.yaml, sexual, and suicide/self-harm harm_definition profiles + malware unit test.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant