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

Folders and files

NameName
Last commit message
Last commit date

Latest commit

207008b · Apr 7, 2022

History

27 Commits
Mar 3, 2022
Mar 3, 2022
Apr 7, 2022
Mar 4, 2022
Apr 7, 2022
Mar 4, 2022
Mar 4, 2022
Mar 4, 2022
Mar 4, 2022
Mar 3, 2022
Apr 7, 2022
Mar 3, 2022

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