ivanlau
/

distil-bert-uncased-finetuned-github-issues

Text Classification

Inference Endpoints

Model card Files Files and versions Community

YAML Metadata Error: "model-index[0].results[0].dataset.type" with value "ticket tagger" fails to match the required pattern: /^(?:[\w-]+\/)?[\w-.]+$/

Model Description

This model is a fine-tuned version of distilbert-base-uncased and fine-tuning it on the github ticket tagger dataset. It classifies issue into 3 common categories: Bug, Enhancement, Questions.

It achieves the following results on the evaluation set:

Accuracy: 0.7862

Training hyperparameters

The following hyperparameters were used during training:

learning_rate: 3e-5
train_batch_size: 16
optimizer: AdamW with betas=(0.9,0.999) and epsilon=1e-08
lr_scheduler_type: linear
lr_scheduler_warmup_steps: 0
num_epochs: 5

Codes

https://github.com/IvanLauLinTiong/IntelliLabel

Downloads last month: 27

Inference Examples

Text Classification

This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

Space using ivanlau/distil-bert-uncased-finetuned-github-issues 1

Evaluation results

Model card error

This model's model-index metadata is invalid: Schema validation error. "model-index[0].results[0].dataset.type" with value "ticket tagger" fails to match the required pattern: /^(?:[\w-]+\/)?[\w-.]+$/