kanalizer-model / model /README.md
sevenc-nanashi's picture
feat: v4
7863e74
|
raw
history blame
987 Bytes

v4

学習

学習側の設定
# 学習に使うデータ。
train_data: ./vendor/v3.jsonl
# 評価に使うデータ。
eval_data: ./vendor/unidic_words.jsonl
# 評価用データのうち、学習中に評価に使う単語の数。(eval.pyでは全てのデータを使います)
# nullの場合は全てのデータを使います。
eval_max_words: 100
# モデルの次元数。
dim: 256
# 学習するエポック数。
max_epochs: 30
# 最新のエポックから保存するモデルの数。
num_last_models_to_keep: 2
# 保存する上位のモデルの数。
num_best_models_to_keep: 2
# 乱数のシード。
seed: 0

optimizer_lr: 0.001
exponential_lr_scheduler_gamma: 0.9