Downloads · 30 days
0
YUGOROU/shiin-ime-gru
shiin-ime-gru is a translation model from YUGOROU. Use it when you need text moved from one language to another. It is set up for pytorch. The card lists the license as mit.
Downloads · 30 days
0
Access
Public
Updated Jun 11, 2026
Repo size
12.3 MB
Likes
0
Public
Click a slice to open those files.
.pt12.3 MB · 100%
From the Hugging Face model README
⚠️ Early version / 初期バージョン This is an early baseline checkpoint (v2). Large-scale improvements to the architecture, training data, and accuracy are planned. これは初期のベースライン(v2)です。アーキテクチャ・学習データ・精度の 大規模な改善を今後予定しています。
shiin-ime-gru is the on-device conversion model behind Shiin IME — an iOS
custom keyboard that lets users type Japanese using consonants only, with no
vowels. The model maps a consonant string to its reading.
Example:
wtsh→watashi(わたし) /tbt→ 食べた /gks→ 学生
| Item | Value |
|---|---|
| Type | Transformer Encoder + GRU Decoder (attention-based seq2seq, character-level) |
| Parameters | ~1.3M (hidden=256, embed=64, heads=4) |
| Input → Output | consonant sequence → reading (romaji / kana) |
| Checkpoint | model_best.pt (PyTorch, ~12 MB) |
| On-device | converted to CoreML (encoder + decoder, ~7 MB total) |
Trained on a mixture of Japanese corpora — Wikipedia, CC-100, and the livedoor news corpus — for ~1B samples over 10 epochs.
| Version | Best CER | Notes |
|---|---|---|
| v2 (this checkpoint) | 0.2160 | epoch 3/10, lr=1e-3 — early baseline |
CER (Character Error Rate) is reported on the validation split. As an early version, accuracy is expected to improve substantially in future releases.
Planned improvements include a larger / revised model architecture, expanded and cleaner training data, and substantially lower CER. Interface-changing updates may require training from scratch rather than fine-tuning this checkpoint.
shiin-ime-gru は、Shiin IME(子音のみIME) のオンデバイス変換モデルです。
Shiin IME は、母音を入力せず 子音のみ で日本語を入力できる iOS 用カスタム
キーボードです。本モデルは子音列をその読みへ変換します。
例:
wtsh→watashi(わたし)/tbt→ 食べた /gks→ 学生
| 項目 | 内容 |
|---|---|
| 種別 | Transformer Encoder + GRU Decoder(attention付き seq2seq・文字レベル) |
| パラメータ | 約1.3M(hidden=256, embed=64, heads=4) |
| 入出力 | 子音列 → 読み(ローマ字 / かな) |
| チェックポイント | model_best.pt(PyTorch, 約12MB) |
| オンデバイス | CoreML へ変換(encoder + decoder, 合計 約7MB) |
日本語コーパス(Wikipedia・CC-100・livedoorニュースコーパス)の混合データを用い、 約10億サンプル・10エポックで学習しています。
| バージョン | Best CER | 備考 |
|---|---|---|
| v2(本チェックポイント) | 0.2160 | epoch 3/10, lr=1e-3 — 初期ベースライン |
CER(文字誤り率)は検証データでの値です。初期バージョンのため、今後のリリースで 精度は大幅に向上する見込みです。
今後、モデルの大型化・アーキテクチャ刷新、学習データの拡充とクリーニング、 CERの大幅な低減を予定しています。入出力インターフェースを変更する更新では、 本チェックポイントからの継続学習ではなくスクラッチからの再学習が必要になる 場合があります。