Views
No views yet
NHSQLNL은 한국어 금융 자연어 질의를 SQL 쿼리로 변환하는 Text-to-SQL (NL2SQL) 모델입니다.transformers 기반1from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
2
3# 모델 로드
4MODEL_PATH = "combe4259/NHSQLNL"
5tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH)
6model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_PATH)
7
8# 입력 질의
9query = "2023년에 개설된 예금 계좌 수를 알려줘"
10
11inputs = tokenizer(query, return_tensors="pt")
12
13# SQL 예측
14outputs = model.generate(**inputs, max_length=128)
15sql = tokenizer.decode(outputs[0], skip_special_tokens=True)
16
17print("입력:", query)
18print("생성된 SQL:", sql)
19
20
21---
22
23## 학습 데이터 (Training Data)
24
25- 자체 구축한 금융 도메인 **자연어 ↔ SQL 매핑 데이터셋** 사용
26- 데이터 전처리: SQL 스키마 정규화 및 토크나이저 기반 입력 변환
27
28---
29---
30
31## 활용 가능 분야 (Applications)
32
33- 금융권 챗봇 및 상담 자동화
34- 자연어 기반 데이터 조회 및 리포트 생성
35- 비전문가 대상 SQL 학습/연습 도구