Add unlabeled-corpus extractor (B-2)

build_unlabeled_corpus.py pulls the 34,030 raw-source dialogues that
never got speech_act/slot labels (found via the earlier QA
cross-check) into their own unlabeled.jsonl, forward-filling the
per-dialogue metadata that AI-Hub's CSV export only writes on each
dialogue's first row. Verified against the actual data: 176,605
raw - 142,575 labeled = 34,030, matches exactly.
This commit is contained in:
Claude 2026-07-29 08:37:18 +00:00
parent 6d9738e153
commit 7307cb56bb
No known key found for this signature in database
3 changed files with 123 additions and 3 deletions

View File

@ -123,6 +123,8 @@
- [x] PoC #1 기반 코퍼스 확보 및 전처리 파이프라인 — AI-Hub "한국어 SNS 멀티턴 대화" 14.2만 건
확보, `poc/tone-corpus/prepare_dataset.py`로 학습/검증 JSONL 정제 완료 (개인화 검증 자체는
아직 별개로 필요 — `poc-plan.md` "필요한 것" 참고)
- [x] 라벨 없는 원천 대화 34,030건 추가 확보 — `poc/tone-corpus/build_unlabeled_corpus.py`
`unlabeled.jsonl` 정리 (화행/슬롯 라벨 없음, 순수 언어모델링용)
- [x] 응답 초안 생성기 프로토타입 — `poc/tone-corpus/generate_draft.py` (말투 예시 + 대화 맥락 →
LLM 호출로 초안 생성, 에스컬레이션 케이스는 `[ESCALATE]`로 거절). API 키 없이 코퍼스 실제
대화로 프롬프트 구성까지만 확인함 — 실제 자동 호출은 `ANTHROPIC_API_KEY` 설정 후 가능

View File

@ -24,9 +24,13 @@ python3 prepare_dataset.py --input <원본 zip이 있는 디렉토리> --output
- 주제 9종, 정치(3,242건)·경제및사회(4,257건)가 다른 주제(1.7만~2.2만건)보다 훨씬 적음 — 주제별
균형이 필요하면 샘플링 시 감안할 것
- 화자 연령/성별 필드 노이즈 81건은 정제 과정에서 `null`로 치환 (버리지 않고 필드만 비움)
- **QA 결과**: 원천(CSV) 쪽에는 라벨링된 142,575건 외에 **34,030건이 더 있음** (라벨 없이 텍스트만).
화행/슬롯 라벨이 필요 없는 순수 언어모델링 목적이라면 이 34,030건도 추가로 끌어올 수 있다 —
다만 이번 스크립트는 라벨 JSON만 다루므로 추가 파서가 필요함 (아직 안 만듦)
- **QA 결과**: 원천(CSV) 쪽에는 라벨링된 142,575건 외에 34,030건이 더 있었음 (라벨 없이 텍스트만).
`build_unlabeled_corpus.py`로 그 34,030건만 뽑아 `unlabeled.jsonl`(63MB)로 정리함 — 화행/슬롯
없이 화자·발화 텍스트만 있어서 순수 언어모델링(다음 문장 예측 등)용으로만 쓸 것
```bash
python3 build_unlabeled_corpus.py --input <원본 zip 디렉토리> --output <출력 디렉토리>
```
## 응답 초안 생성기 (`generate_draft.py`)

View File

@ -0,0 +1,114 @@
#!/usr/bin/env python3
"""Pull the dialogues that exist in the AI-Hub raw source (VS_/TS_ CSV) but
were never labeled (VL_/TL_ JSON) -- 34,030 of them, per the --qa check in
prepare_dataset.py. No speech_act/slot labels are available for these, so
they're only useful for plain language-modeling, not the labeled fields
`generate_draft.py`'s prompt doesn't use anyway.
Usage:
python3 build_unlabeled_corpus.py --input <dir with the *.zip parts> --output <output dir>
"""
import argparse
import csv
import glob
import io
import json
import os
import zipfile
from collections import OrderedDict
VALID_SEX = {"남자", "여자"}
def is_valid_age(value):
return bool(value) and value.isdigit() and 10 <= int(value) <= 90 and int(value) % 10 == 0
def clean(value, kind):
if not value:
return None
if kind == "sex":
return value if value in VALID_SEX else None
return value if is_valid_age(value) else None
def labeled_ids(input_dir):
ids = set()
for path in glob.glob(os.path.join(input_dir, "*VL_*.zip")) + glob.glob(
os.path.join(input_dir, "*TL_*.zip")
):
with zipfile.ZipFile(path) as z:
for name in z.namelist():
if name.endswith(".json"):
with z.open(name) as f:
ids.add(json.load(f)["info"]["id"])
return ids
def read_csv_dialogues(input_dir):
dialogues = OrderedDict()
csv_paths = sorted(
glob.glob(os.path.join(input_dir, "*VS_*.zip")) + glob.glob(os.path.join(input_dir, "*TS_*.zip"))
)
for path in csv_paths:
with zipfile.ZipFile(path) as z:
for name in z.namelist():
if not name.endswith(".csv"):
continue
with z.open(name) as f:
text = io.TextIOWrapper(f, encoding="utf-8-sig")
current = None
for row in csv.DictReader(text):
if row.get("대화ID"):
did = row["대화ID"]
speakers = {}
for letter in ("A", "B", "C"):
sid = row.get(f"화자{letter} ID")
if not sid:
continue
speakers[letter] = {
"id": sid,
"sex": clean(row.get(f"화자{letter} 성별"), "sex"),
"age": clean(row.get(f"화자{letter} 연령대"), "age"),
}
current = {
"id": did,
"topic": row.get("주제"),
"keyword": row.get("키워드"),
"speakers": speakers,
"utterances": [],
}
dialogues[did] = current
if current is None:
continue
current["utterances"].append(
{"speaker": row.get("발화자", ""), "text": row.get("발화", "")}
)
return dialogues
def main():
ap = argparse.ArgumentParser()
ap.add_argument("--input", required=True)
ap.add_argument("--output", required=True)
args = ap.parse_args()
os.makedirs(args.output, exist_ok=True)
already_labeled = labeled_ids(args.input)
all_dialogues = read_csv_dialogues(args.input)
out_path = os.path.join(args.output, "unlabeled.jsonl")
kept = 0
with open(out_path, "w", encoding="utf-8") as out:
for did, d in all_dialogues.items():
if did in already_labeled:
continue
d["turns"] = len(d["utterances"])
out.write(json.dumps(d, ensure_ascii=False) + "\n")
kept += 1
print(f"원천 전체: {len(all_dialogues)}건 / 라벨링됨(제외): {len(already_labeled)}건 / 출력: {kept}건 -> {out_path}")
if __name__ == "__main__":
main()