Fewshotissueclassifier NLBSE23

🚀 FewShotIssueClassifier-NLBSE23

このモデルは、Sentence Transformersを使用して文や段落を768次元の密なベクトル空間にマッピングするSetFitモデルです。クラスタリングや意味検索などのタスクに使用できます。

この特定のモデルは、Issue Report Classificationのために4つのクラス（バグ、ドキュメント、機能、質問）に微調整されています。

🚀 クイックスタート

このモデルは、文や段落を768次元の密なベクトル空間にマッピングするためにSentence Transformersを使用したSetFitモデルです。クラスタリングや意味検索などのタスクに利用できます。

✨ 主な機能

• Sentence Transformersを使用して文や段落を768次元の密なベクトル空間にマッピングします。
• 4つのクラス（バグ、ドキュメント、機能、質問）に対するIssue Report Classificationに微調整されています。

💻 使用例

基本的な使用法

from sentence_transformers.losses import CosineSimilarityLoss
from setfit import SetFitModel
from setfit import SetFitTrainer
sentences = ["error in line 20", "add method list_features"]

label_mapping = {
  0 : "bug",
  1 : "documentation",
  2 : "feature",
  3 : "question"
}

model = SetFitModel.from_pretrained('PeppoCola/FewShotIssueClassifier-NLBSE23')
predictions = model.predict(sentences)
print([label_mapping[i] for i in predictions])

📚 ドキュメント

データセット

このモデルは、NLBSE23データセットのサブセットで学習されています。サンプルは手動でラベル付けされ、Zenodoで公開されています。

引用と作者

@software{Colavito_Few-Shot_Learning_for_2023,
	title        = {{Few-Shot Learning for Issue Report Classification}},
	author       = {Colavito, Giuseppe and Lanubile, Filippo and Novielli, Nicole},
	year         = 2023,
	month        = 2,
	url          = {https://github.com/collab-uniba/Issue-Report-Classification-NLBSE2023},
	version      = {1.0.0}
}

@dataset{colavito_giuseppe_2023_7628150,
  author       = {Colavito Giuseppe and
                  Lanubile Filippo and
                  Novielli Nicole},
  title        = {Few-Shot Learning for Issue Report Classification},
  month        = feb,
  year         = 2023,
  note         = {{To use this, merge the CSV with the original 
                   dataset (after removing duplicates on the 'id'
                   column)}},
  publisher    = {Zenodo},
  doi          = {10.5281/zenodo.7628150},
  url          = {https://doi.org/10.5281/zenodo.7628150}
}

@inproceedings{Colavito-2023,
	title        = {Few-Shot Learning for Issue Report Classification},
	author       = {Colavito, Giuseppe and Lanubile, Filippo and Novielli, Nicole},
	year         = 2023,
	booktitle    = {2nd International Workshop on Natural Language-Based Software Engineering (NLBSE)}
}

📄 ライセンス

このモデルはGPL-3.0ライセンスの下で提供されています。