커스텀 파이프라인 훈련

커스텀 파이프라인 훈련

도메인에 맞는 개체명이나 카테고리를 추가하고 싶다면 spaCy로 직접 훈련할 수 있어요. spaCy는 설정 파일(config.cfg) 기반으로 훈련을 진행해요.

훈련 흐름

  1. 학습·검증 데이터를 준비하고 .spacy 형식으로 변환
  2. init config로 설정 파일 생성
  3. train 명령으로 훈련
python -m spacy init config config.cfg --lang en --pipeline ner
python -m spacy train config.cfg --output ./output

확인 포인트

  • NER처럼 컴포넌트는 --pipeline ner로 지정해요.
  • 훈련 데이터는 DocBin로 .spacy 파일로 만들어야 해요.

출처: https://spacy.io/usage/training

더 알아보기