Skip to content

Files

Latest commit

664b46d · Sep 1, 2022

History

History
14 lines (10 loc) · 939 Bytes

README.md

File metadata and controls

14 lines (10 loc) · 939 Bytes

2nd place solution in the Efficiency Track

Kaggle Feedback Prize - Predicting Effective Arguments

This repository contains the code that was used to train the 2nd place solution in the Efficiency Track of Kaggle Feedback Prize - Predicting Effective Arguments competition.

We followed a multi-stage approach to training the winning model, so the exact score may be hard to reproduce. I will highlight below the key elements of the solution:

  1. MLM pre-training: notebooks/HF-pret-7.py
  2. 1st stage models: notebooks/HF-43.ipynb
  3. 2nd stage model distillation on pseudolabels: notebooks/HF-56-pseudolabels.ipynb
  4. 2nd stage model finetuning: notebooks/HF-43pseudo2.ipynb