Repolode
제출하기
pyg-team

pyg-team/pytorch-frame

텍스트와 이미지 등 여러 유형의 열이 섞인 표 형식 데이터에서 딥러닝 모델을 구축하고 학습할 수 있도록 PyTorch를 확장합니다.

GitHub에서 보기웹사이트
스타
802
포크
75
열린 이슈
25
마지막 커밋
4일 전

어떤 프로젝트인가요?

PyTorch Frame은 열마다 수치형, 범주형, 다중 범주형, 타임스탬프, 텍스트, 이미지 또는 임베딩 값이 들어갈 수 있는 이기종 표 형식 데이터를 위한 PyTorch 딥러닝 확장 라이브러리입니다. pandas DataFrame을 tensor frame으로 변환하고, 열 유형별로 인코딩한 다음, 열 간 상호작용을 모델링하고 행별 예측을 디코딩하는 모듈형 파이프라인을 제공합니다.

FTTransformer, ResNet, TabNet, TabTransformer, Trompt, ExcelFormer 등 여러 심층 표 데이터 아키텍처를 구현했습니다. 또한 하이퍼파라미터 튜닝에 Optuna를 사용하는 XGBoost, CatBoost, LightGBM 예제도 제공합니다. 바로 사용할 수 있는 벤치마크 데이터셋이 포함되어 있으며, 사용자 지정 데이터도 지원합니다. 텍스트 열은 OpenAI, Cohere, Voyage AI 또는 Hugging Face 모델의 임베딩으로 인코딩할 수 있습니다.

PyTorch를 기반으로 구축되며 PyTorch Geometric과 함께 사용할 수 있어, 관계형 데이터베이스 학습과 같은 작업에서 다른 PyTorch 모델과 결합할 수 있습니다.

핵심 기능

  • 01수치형, 범주형, 다중 범주형, 타임스탬프, 텍스트, 이미지 및 임베딩 열을 지원합니다.
  • 02FTTransformer, ResNet, TabNet, TabTransformer 등 여러 심층 표 데이터 모델을 구현합니다.
  • 03Optuna 하이퍼파라미터 튜닝을 사용하는 GBDT 예제(XGBoost, CatBoost, LightGBM)를 제공합니다.
  • 04사용자 지정 모델을 구축할 수 있도록 모듈형 인코더, 컨볼루션, 디코더 구조를 제공합니다.
  • 05내장 표 데이터셋과 미니배치 데이터 로더를 포함합니다.
  • 06텍스트 열을 위해 임베딩 API 및 Hugging Face 모델과 연동합니다.

활용 사례

  • 숫자, 범주, 자유 형식 텍스트 필드가 섞인 비즈니스 데이터셋으로 신경망을 학습합니다.
  • 표준 벤치마크에서 새로운 심층 표 데이터 아키텍처를 GBDT 기준 모델과 비교합니다.
  • LLM이 생성한 텍스트 임베딩을 표 형식 예측 모델의 특성으로 추가합니다.
  • PyTorch를 기반으로 사용자 지정 표 데이터 모델 아키텍처를 구축합니다.

사용 전 알아둘 점

  • README 발췌문에 설치 안내가 없어 여기에 설치 단계를 나열하지 않았습니다.
  • 일부 텍스트 임베딩 예제는 OpenAI, Cohere, Voyage AI와 같은 유료 외부 임베딩 API를 사용합니다.

토픽

#deep-learning#pytorch#tabular-data#tabular-learning#machine-learning#embeddings#data-frame

자주 묻는 질문

pytorch-frame은(는) 무엇인가요?
텍스트와 이미지 등 여러 유형의 열이 섞인 표 형식 데이터에서 딥러닝 모델을 구축하고 학습할 수 있도록 PyTorch를 확장합니다. PyTorch Frame은 열마다 수치형, 범주형, 다중 범주형, 타임스탬프, 텍스트, 이미지 또는 임베딩 값이 들어갈 수 있는 이기종 표 형식 데이터를 위한 PyTorch 딥러닝 확장 라이브러리입니다. pandas DataFrame을 tensor frame으로 변환하고, 열 유형별로 인코딩한 다음, 열 간 상호작용을 모델링하고 행별 예측을 디코딩하는 모듈형 파이프라인을 제공합니다.
pytorch-frame은(는) 무료로 사용할 수 있나요? 라이선스는 무엇인가요?
pytorch-frame은(는) MIT 라이선스의 오픈소스입니다. 상업용 제품에 사용하기 전에 라이선스 조건을 확인하세요.
pytorch-frame은(는) 아직 유지보수되고 있나요?
활발히 유지보수 중. pytorch-frame의 마지막 커밋은 2026년 11월 2일입니다.
pytorch-frame은(는) 어디에 사용할 수 있나요?
숫자, 범주, 자유 형식 텍스트 필드가 섞인 비즈니스 데이터셋으로 신경망을 학습합니다. 표준 벤치마크에서 새로운 심층 표 데이터 아키텍처를 GBDT 기준 모델과 비교합니다. LLM이 생성한 텍스트 임베딩을 표 형식 예측 모델의 특성으로 추가합니다. PyTorch를 기반으로 사용자 지정 표 데이터 모델 아키텍처를 구축합니다.
pytorch-frame은(는) 어디서 받을 수 있나요?
소스 코드와 설치 방법은 GitHub에 있습니다: https://github.com/pyg-team/pytorch-frame

요약은 프로젝트 README를 바탕으로 AI가 작성했습니다. GitHub 통계는 매일 갱신됩니다.

관련 프로젝트