지도 학습 텍스트 분류

지도 학습 텍스트 분류

fastText는 텍스트 분류기도 빠르게 학습할 수 있어요. 각 줄이 __label__로 시작하는 라벨과 텍스트로 구성된 형식의 데이터로 train_supervised()를 호출하면 돼요.

데이터 형식

__label__positive This movie is great.
__label__negative I did not like this film.

학습·평가

import fasttext

model = fasttext.train_supervised(input="train.txt", epoch=25, lr=1.0)
print(model.test("valid.txt"))

# 하이퍼파라미터 자동 탐색
model = fasttext.train_supervised(input="train.txt", autotuneValidationFile="valid.txt")

확인 포인트

  • epoch, lr 같은 하이퍼파라미터가 성능에 큰 영향을 줘요.
  • autotuneValidationFile을 주면 자동으로 하이퍼파라미터를 탐색해요.

출처: https://fasttext.cc/docs/en/supervised-models.html

더 알아보기