microsoft/hummingbird

어떤 프로젝트인가요?

핵심 기능

• 01scikit-learn 트리, 랜덤 포레스트, LightGBM 및 XGBoost 모델을 변환합니다
• 02PyTorch, TorchScript, ONNX 및 TVM 백엔드를 대상으로 합니다
• 03변환된 모델에 scikit-learn 스타일의 predict API를 제공합니다
• 04model.to('cuda')를 사용하는 경우를 포함해 CPU 또는 GPU에서 예측을 실행합니다
• 05변환된 모델의 저장과 불러오기를 지원합니다

활용 사례

• GPU 하드웨어에서 학습된 랜덤 포레스트의 배치 추론 속도 높이기
• PyTorch 기반의 단일 서빙 스택에서 기존 ML 모델과 신경망을 함께 배포하기
• Python 외부에서 실행할 수 있도록 트리 기반 모델을 ONNX 또는 TVM으로 내보내기

빠른 시작

1. 1

   pytorch.org에서 사용 중인 플랫폼에 맞는 PyTorch(버전 1.6.0 이상)를 설치합니다

2. 2

   pip에서 Hummingbird를 설치합니다

   python -m pip install hummingbird-ml
3. 3

   hummingbird.ml을 가져온 후 convert(model, 'pytorch')로 학습된 scikit-learn 모델을 변환합니다

사용 전 알아둘 점

• !Hummingbird를 사용하려면 먼저 PyTorch를 별도로 설치해야 합니다.
• !TVM 백엔드는 Python 3.10에서만 작동합니다.
• !scikit-learn, LightGBM, XGBoost 모델과 피처 변환기 중 일부만 지원하며, 모두 지원하지는 않습니다.

포함된 컬렉션

토픽

자주 묻는 질문

hummingbird은(는) 무엇인가요?

학습된 scikit-learn, LightGBM, XGBoost 모델을 텐서 연산으로 변환해 PyTorch, TorchScript, ONNX 또는 TVM에서 추론을 실행하는 Python 라이브러리입니다. Hummingbird는 학습된 기존 머신러닝 모델을 텐서 연산으로 변환해 신경망 프레임워크에서 실행할 수 있게 합니다. 개발자는 scikit-learn 의사결정 트리와 랜덤 포레스트, LightGBM 또는 XGBoost 분류기와 회귀 모델을 가져와 모델을 다시 작성하지 않고도 PyTorch, TorchScript, ONNX 또는 TVM을 대상으로 변환할 수 있습니다.

hummingbird은(는) 무료로 사용할 수 있나요? 라이선스는 무엇인가요?

hummingbird은(는) MIT 라이선스의 오픈소스입니다. 상업용 제품에 사용하기 전에 라이선스 조건을 확인하세요.

hummingbird은(는) 아직 유지보수되고 있나요?

6개월 이내 업데이트. hummingbird의 마지막 커밋은 2026년 9월 14일입니다.

hummingbird은(는) 어디에 사용할 수 있나요?

GPU 하드웨어에서 학습된 랜덤 포레스트의 배치 추론 속도 높이기 PyTorch 기반의 단일 서빙 스택에서 기존 ML 모델과 신경망을 함께 배포하기 Python 외부에서 실행할 수 있도록 트리 기반 모델을 ONNX 또는 TVM으로 내보내기

hummingbird는 어떻게 설치하나요?

pytorch.org에서 사용 중인 플랫폼에 맞는 PyTorch(버전 1.6.0 이상)를 설치합니다 pip에서 Hummingbird를 설치합니다: python -m pip install hummingbird-ml hummingbird.ml을 가져온 후 convert(model, 'pytorch')로 학습된 scikit-learn 모델을 변환합니다

hummingbird은(는) 어디서 받을 수 있나요?

소스 코드와 설치 방법은 GitHub에 있습니다: https://github.com/microsoft/hummingbird