🚀 コードドキュメント生成用JavaScriptのCodeTransモデル
t5 smallモデルアーキテクチャを使用して、プログラミング言語JavaScriptで事前学習されたモデルです。このモデルは、このリポジトリで最初に公開されました。このモデルはトークン化されたJavaScriptコード関数で学習されており、トークン化されたJavaScript関数で最も良い性能を発揮します。
🚀 クイックスタート
このモデルは、JavaScript関数の説明を生成するために使用でき、他のJavaScriptコードタスクで微調整することもできます。解析されていない、トークン化されていないJavaScriptコードでも使用できますが、JavaScriptコードがトークン化されている場合、性能は向上するはずです。
✨ 主な機能
- JavaScript関数の説明を生成することができます。
- 他のJavaScriptコードタスクで微調整することができます。
📦 インストール
このモデルを使用するには、transformers
ライブラリが必要です。以下のコマンドでインストールできます。
pip install transformers
💻 使用例
基本的な使用法
以下は、TransformersのSummarizationPipelineを使用してこのモデルを使ってJavaScript関数のドキュメントを生成する方法です。
from transformers import AutoTokenizer, AutoModelWithLMHead, SummarizationPipeline
pipeline = SummarizationPipeline(
model=AutoModelWithLMHead.from_pretrained("SEBIS/code_trans_t5_small_code_documentation_generation_javascript"),
tokenizer=AutoTokenizer.from_pretrained("SEBIS/code_trans_t5_small_code_documentation_generation_javascript", skip_special_tokens=True),
device=0
)
tokenized_code = "function isStandardBrowserEnv ( ) { if ( typeof navigator !== 'undefined' && ( navigator . product === 'ReactNative' || navigator . product === 'NativeScript' || navigator . product === 'NS' ) ) { return false ; } return ( typeof window !== 'undefined' && typeof document !== 'undefined' ) ; }"
pipeline([tokenized_code])
この例をcolab notebookで実行できます。
📚 ドキュメント
モデルの説明
このCodeTransモデルは、t5-small
モデルに基づいています。独自のSentencePiece語彙モデルを持っており、CodeSearchNet CorpusのJavaScriptデータセットで単一タスク学習を行っています。
想定される用途と制限
このモデルは、JavaScript関数の説明を生成するために使用でき、他のJavaScriptコードタスクで微調整することもできます。解析されていない、トークン化されていないJavaScriptコードでも使用できますが、JavaScriptコードがトークン化されている場合、性能は向上するはずです。
🔧 技術詳細
このモデルは、トークン化されたJavaScriptコード関数で学習されています。トークン化されたJavaScript関数で最も良い性能を発揮します。
📄 ライセンス
このモデルの作成者は、Ahmed Elnaggar | LinkedIn と Wei Ding | LinkedIn です。
📦 学習データ
教師付き学習タスクのデータセットは、こちらのリンクからダウンロードできます。
📊 評価結果
コードドキュメントタスクにおいて、異なるモデルが異なるプログラミング言語で次のような結果(BLEUスコア)を達成しています。
言語 / モデル |
Python |
Java |
Go |
Php |
Ruby |
JavaScript |
CodeTrans-ST-Small |
17.31 |
16.65 |
16.89 |
23.05 |
9.19 |
13.7 |
CodeTrans-ST-Base |
16.86 |
17.17 |
17.16 |
22.98 |
8.23 |
13.17 |
CodeTrans-TF-Small |
19.93 |
19.48 |
18.88 |
25.35 |
13.15 |
17.23 |
CodeTrans-TF-Base |
20.26 |
20.19 |
19.50 |
25.84 |
14.07 |
18.25 |
CodeTrans-TF-Large |
20.35 |
20.06 |
19.54 |
26.18 |
14.94 |
18.98 |
CodeTrans-MT-Small |
19.64 |
19.00 |
19.15 |
24.68 |
14.91 |
15.26 |
CodeTrans-MT-Base |
20.39 |
21.22 |
19.43 |
26.23 |
15.26 |
16.11 |
CodeTrans-MT-Large |
20.18 |
21.87 |
19.38 |
26.08 |
15.00 |
16.23 |
CodeTrans-MT-TF-Small |
19.77 |
20.04 |
19.36 |
25.55 |
13.70 |
17.24 |
CodeTrans-MT-TF-Base |
19.77 |
21.12 |
18.86 |
25.79 |
14.24 |
18.62 |
CodeTrans-MT-TF-Large |
18.94 |
21.42 |
18.77 |
26.20 |
14.19 |
18.83 |
最先端技術 |
19.06 |
17.65 |
18.07 |
25.16 |
12.16 |
14.90 |