HW-202337458-bert-imdb-finetuning

Assignment

05월 28일 파인튜닝 실습 노트북 업로드 과제입니다.

Student

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

Contents

• IMDB 전체 데이터셋으로 BERT 파인튜닝 수행
• Google Drive 연동
• 날짜/시간 기반 output_dir 자동 생성
• 체크포인트 저장
• 중단 후 이어서 학습 재개
• Accuracy, Precision, Recall, F1 출력
• 학습 커브 시각화

Files

• bert_imdb_finetuning_executed.ipynb
• final_metrics.json
• trainer_log_history.csv
• loss_curve.png
• metric_curve.png
• README.md

Output Directory

/content/drive/MyDrive/bert_imdb_finetuning_outputs/20260610_153314

Validation Metrics

{
  "eval_loss": 0.32361480593681335,
  "eval_accuracy": 0.8844,
  "eval_precision": 0.8835182250396196,
  "eval_recall": 0.8870326173428799,
  "eval_f1": 0.8852719333068678,
  "eval_runtime": 7.4983,
  "eval_samples_per_second": 333.41,
  "eval_steps_per_second": 41.743,
  "epoch": 1.0
}

Test Metrics

{
  "eval_loss": 0.3114667236804962,
  "eval_accuracy": 0.88444,
  "eval_precision": 0.8705945862574227,
  "eval_recall": 0.90312,
  "eval_f1": 0.8865590764518789,
  "eval_runtime": 70.3693,
  "eval_samples_per_second": 355.268,
  "eval_steps_per_second": 44.409,
  "epoch": 1.0
}

Created At

2026-06-10 16:09:29