Skip to content

Finetuning DistilBert model from HuggingFace to predict Github issues' labels

Notifications You must be signed in to change notification settings

IvanLauLinTiong/IntelliLabel

Repository files navigation

IntelliLabel

app.png

IntelliLabel is built by using pretrained distilbert-base-uncased model and fine-tuning it on the github ticket tagger dataset under PyTorch and HuggingFace framework. It can classify issue into 3 common categories: Bug, Enhancement, Questions.

Clone Project

Run the commands belows to download project along with the model bin from HuggingFace Hub:

git clone https://github.com/IvanLauLinTiong/IntelliLabel.git
cd IntelliLabel
git submodule update --init --recursive

Or simply:

git clone --recurse-submodules https://github.com/IvanLauLinTiong/IntelliLabel.git

Demo

Try it out: Streamlit Cloud OR HuggingFace Spaces

About

Finetuning DistilBert model from HuggingFace to predict Github issues' labels

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published