ktrain 개요

ktrain은 '스위스 아미 나이프'를 표방하는 머신러닝 라이브러리예요. TensorFlow Keras를 가볍게 감싸서, 텍스트·비전·그래프·테이블 데이터를 몇 줄의 코드로 학습하고 배포할 수 있게 해 줘요.

출처: amaiya/ktrain

ktrain은 딥러닝 라이브러리 TensorFlow Keras(및 다른 라이브러리)를 위한 경량 래퍼예요. fastai와 ludwig 같은 ML 프레임워크 확장에서 영감을 받았고, 신경망과 기타 머신러닝 모델을 더 쉽게 만들고 학습하고 배포하는 걸 목표로 해요.

텍스트 데이터로는 BERT·DistilBERT·NBSVM·fastText 기반 분류, 시퀀스 라벨링(NER), 문장 쌍 분류, 추출형·생성형 QA, 요약, 제로샷 학습, 번역, 문서 유사도, 감성 분석 등을 다룰 수 있어요. 비전 데이터로는 이미지 분류·회귀·캡셔닝·객체 검출을, 그래프 데이터로는 GraphSAGE 기반 노드 분류와 링크 예측을 지원해요.

또 학습률 파인더(Learning Rate Finder)로 최적 학습률을 찾고, 1cycle 정책이나 SGDR 같은 학습률 스케줄을 활용할 수 있어요. 훈련 후에는 간단한 예측 API로 모델과 전처리 단계를 함께 저장·배포할 수 있는 것도 큰 장점이에요.