feat: initial CompanionGuard-RL framework
Two-module pipeline for AI companion safety: - Module B: context-aware risk detector with CrossAttention fusion - Module C: PPO-based adaptive intervention policy Includes CompanionRisk Taxonomy (10 primary + 14 fine-grained labels), dataset generation/annotation pipeline, training scripts, and eval suite. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
22
configs/data_generation.yaml
Normal file
22
configs/data_generation.yaml
Normal file
@@ -0,0 +1,22 @@
|
||||
api:
|
||||
type: "qwen" # "qwen" or "openai"
|
||||
model: "qwen-max"
|
||||
|
||||
generation:
|
||||
total_samples: 3000
|
||||
samples_per_category: 300
|
||||
delay: 0.5 # seconds between API calls
|
||||
|
||||
output:
|
||||
raw_dir: "data/raw"
|
||||
output_file: "data/raw/generated.jsonl"
|
||||
|
||||
annotation:
|
||||
judge_model: "qwen-max"
|
||||
output_file: "data/processed/annotated.jsonl"
|
||||
|
||||
split:
|
||||
train: 0.8
|
||||
val: 0.1
|
||||
test: 0.1
|
||||
seed: 42
|
||||
Reference in New Issue
Block a user