fastText Python API

fastText Python API

fastText는 파이썬에서 아주 간단한 인터페이스로 단어 벡터와 분류기를 다룰 수 있어요. fasttext 패키지를 설치하면 load_model(), get_word_vector() 같은 함수를 바로 쓸 수 있죠.

단어 벡터 예시

import fasttext

model = fasttext.load_model("cc.en.300.bin")
vec = model.get_word_vector("dog")
print(model.get_nearest_neighbors("dog"))

분류 모델 예시

model = fasttext.train_supervised(input="train.txt")
print(model.predict("Which programming language is C++?"))

확인 포인트

  • predict()의 두 번째 반환값으로 라벨별 확률도 얻을 수 있어요.
  • 모델 파일(.bin)이 있어야 벡터·분류 모두 가능해요.

출처: https://fasttext.cc/docs/en/python-API.html

더 알아보기