Skip to content

Add DeBERTa training and ONNX export pipeline for required phrases extraction - #5222

Open
Kaushik-Kumar-CEG wants to merge 1 commit into
aboutcode-org:developfrom
Kaushik-Kumar-CEG:gsoc/train-deberta-model
Open

Add DeBERTa training and ONNX export pipeline for required phrases extraction#5222
Kaushik-Kumar-CEG wants to merge 1 commit into
aboutcode-org:developfrom
Kaushik-Kumar-CEG:gsoc/train-deberta-model

Conversation

@Kaushik-Kumar-CEG

@Kaushik-Kumar-CEG Kaushik-Kumar-CEG commented Jul 5, 2026

Copy link
Copy Markdown

Part of #5137.

Adds the training and ONNX export scripts for the required phrase model. The model uses DeBERTa with BIOES labels and a CRF layer, and keeps the checkpoint with the best validation F1.

Also includes focused tests for dataset validation, token alignment, model loading, and export.

Historical result: 0.8445 test F1. A new full run with the updated script is pending.

The packaged version is in aboutcode-org/scancode-required-phrases#2.

Train the DeBERTa BIOES and CRF model with validated datasets and best-checkpoint selection. Add strict model export, reproducibility metadata, and focused offline tests.

Signed-off-by: Kaushik Kumar <kaushikrjpm10@gmail.com>
@Kaushik-Kumar-CEG
Kaushik-Kumar-CEG force-pushed the gsoc/train-deberta-model branch from 750d2ab to 4ccfe39 Compare August 31, 2026 15:01
@Kaushik-Kumar-CEG
Kaushik-Kumar-CEG marked this pull request as ready for review August 31, 2026 15:01
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant