fnnx-ai/scikit-llm

What it solves

ChatGPT와 같은 대규모 언어 모델(LLMs)을 익숙한 scikit-learn 생태계에 통합하여 텍스트 분석 작업을 수행하는 과정을 단순화하여, 사용자가 표준 머신러닝 pipeline 내에서 LLM의 능력을 활용할 수 있도록 합니다.

How it works

이 라이브러리는 LLM을 위한 scikit-learn 호환 래퍼(wrappers)를 제공합니다. 예를 들어, ZeroShotGPTClassifier를 통해 OpenAI API key를 설정하고 scikit-learn 모델에서 흔히 사용되는 표준 .fit() 및 .predict() 메서드를 사용하여 제로샷(zero-shot) 텍스트 분류를 수행할 수 있습니다.

Who it's for

scikit-learn에 익숙하며 기존 워크플로우를 재작성하지 않고 LLM을 텍스트 분석에 사용하고자 하는 데이터 과학자 및 머신러닝 엔지니어입니다.

Highlights

• scikit-learn API와의 원활한 통합
• GPT 모델을 사용한 제로샷 텍스트 분류 지원
• SKLLMConfig를 통한 간편한 설정