HW-202337458-mytext-sequence-classification

Assignment

커스텀 모델 정리 및 업로드 과제입니다.

Student

• Student ID: 202337458
• Hugging Face ID: hf-june

Included Classes

• MyTextConfig
• MyTextSequenceClassification

Requirements

• MyTextConfig 코드 정리
• MyTextSequenceClassification 코드 정리
• 마크다운 주석 포함
• save_pretrained 사용
• push_to_hub 사용

Files

• modeling_mytext_sequence_classification.py
• mytext_sequence_classification_summary.ipynb
• saved_model/
• test_result.json
• README.md

How to Use

import torch
from modeling_mytext_sequence_classification import MyTextConfig, MyTextSequenceClassification

config = MyTextConfig()
model = MyTextSequenceClassification(config)

input_ids = torch.randint(0, config.vocab_size, (2, 10))
attention_mask = torch.ones_like(input_ids)
labels = torch.tensor([0, 1])

outputs = model(input_ids=input_ids, attention_mask=attention_mask, labels=labels)
print(outputs.loss)
print(outputs.logits)