ysakuramoto commited on
Commit
3ec94b9
1 Parent(s): 00cfab8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -49,7 +49,7 @@ model = MobileBertForSequenceClassification.from_pretrained("ysakuramoto/mobileb
49
  - 条件
50
  - ストックーマークさんのwikipediaデータセットで学習・推論。(https://github.com/stockmarkteam/ner-wikipedia-dataset)
51
  - BERTモデルに、'cl-tohoku/bert-base-japanese-whole-word-masking'を利用。
52
- - 推論データ n=2,140。精度は完全一致のAccuracy
53
  - エポック数=10, lr=1e-4
54
  - 推論時の高速化として、枝刈り・量子化・jitコンパイルを実施。
55
  - Google Colabにて、学習にGPU、推論にCPUを利用。推論はバッチ処理でなく1件ずつ処理。
 
49
  - 条件
50
  - ストックーマークさんのwikipediaデータセットで学習・推論。(https://github.com/stockmarkteam/ner-wikipedia-dataset)
51
  - BERTモデルに、'cl-tohoku/bert-base-japanese-whole-word-masking'を利用。
52
+ - 推論データ n=2,140。精度は完全一致のf-measure
53
  - エポック数=10, lr=1e-4
54
  - 推論時の高速化として、枝刈り・量子化・jitコンパイルを実施。
55
  - Google Colabにて、学習にGPU、推論にCPUを利用。推論はバッチ処理でなく1件ずつ処理。