جمع‌آوری مجموعهٔ بزرگ کد

برای آموزش یک مدل تولید کد پایتون گتپ‑۲، اولین گام جمع‌آوری دادهٔ کافی است. نویسندگان از dump گیت‌هاب در Google BigQuery استفاده کردند و فایلی با حجم ۱۸۰ گیگابایت و ۲۰ میلیون فایل پایتون استخراج کردند. سپس برای بهبود کیفیت مدل، نوشتن کد تکراری حذف و داده‌های پر نویز فیلتر شد.

  • دسترسی به dump گیت‌هاب در BigQuery
  • پایپ‌لاین استخراج فایل‌های پایتون
  • پاک‌سازی، حذف تکراری‌ها و نویز

آغاز شک‌سازی توکنایزر و مدل

در این مرحله توکنایزر خاص برای ساختار کد پایتون طراحی می‌شود؛ نمونه‌سازی متنی GPT‑2 را با پارامترهای سایز دلخواه، تعداد لایه و اهداف نهایی آغاز می‌کنند. تنظیمات اولیه شامل حداکثر توکن ۵۱۲ و وزن‌های تصادفی اولیه است.

دورۀ آموزش، ارزیابی و نتیجهٔ نهایی

از سپس حلقهٔ آموزش سفارشی استفاده می‌شود که بازخورد، لغزش گرادیان و مقیاسی‌ بندی تولید می‌کند. پس از تکمیل ۲۰۰ هزار گام، مدل خروجی با معیارهای BLEU و Perplexity ارزیابی شد و عملکرد مشابه GitHub CoPilot بود. نسخهٔ نهایی در Hugging Face میزبانی می‌شود تا توسط برنامه‌نویسان برای تکمیل خودکار کد مورد استفاده قرار گیرد.

پرسش‌های پرتکرار

چگونه می‌توانم مدل CodeParrot را در محیطی دیگر اجرا کنم؟

می‌توانید مدل را با استفاده از کتابخانهٔ Transformers و PyTorch بارگذاری کرده و در محیط محلی یا سرورهای ابری اجرا کنید؛ مستندات کامل در Hugging Face موجود است.

کدام زبان‌های برنامه‌نویسی را CodeParrot می‌تواند تولید کند؟

مدل آموزش دیده بر روی فایل‌های پایتون است، بنابراین به‌طور خاص برای تولید کد پایتون طراحی شده؛ با تنظیم دقیق می‌توان آن را برای زبان‌های دیگر گسترش داد.