CoolFace
Modelpublic

caffsean/t5-large-finetune-keyword-to-text-generation

sourceHugging Faceapache-2.0updated 4y agoView on Hugging Face
2likes23downloads
Model Card

<!-- This model card has been generated automatically according to the information the Trainer had access to. You should probably proofread and complete it, then remove this comment. -->

t5-large-finetune-keyword-to-text-generation

This model is a fine-tuned version of t5-large on an unknown dataset. It achieves the following results on the evaluation set:

  • —Loss: 3.1471
  • —Rouge1: 2.175
  • —Rouge2: 0.3661
  • —Rougel: 1.7927
  • —Rougelsum: 1.7951
  • —Gen Len: 15.3252

Model description

This model is designed to generate text from a single keyword. This project is intended to be used for generating vocabulary questions for ed-tech applications.

NOTE!: Be sure to use the 'summarize: ' prefix before the word that you would like to un-summarize.

Intended uses & limitations

More information needed

Training and evaluation data

More information needed

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

  • —learning_rate: 2e-05
  • —trainbatchsize: 8
  • —evalbatchsize: 8
  • —seed: 42
  • —optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • —lrschedulertype: linear
  • —num_epochs: 10

Training results

Training LossEpochStepValidation LossRouge1Rouge2RougelRougelsumGen Len
3.30831.030003.17062.14980.3311.75791.76116.6826
3.21212.060003.14032.15550.34091.76591.76916.208
3.12863.090003.13002.15770.35111.77031.773315.9009
3.05674.0120003.12822.1830.35841.78951.790915.7135
2.99535.0150003.12932.15890.35251.7761.778115.678
2.94836.0180003.13082.16450.35561.78241.78415.425
2.90097.0210003.13582.16220.36221.78481.787715.3348
2.87528.0240003.13872.17160.361.79361.796315.5296
2.8359.0270003.14542.18060.36581.79411.796615.4625
2.835210.0300003.14712.1750.36611.79271.795115.3252

Framework versions

  • —Transformers 4.21.2
  • —Pytorch 1.12.1+cu113
  • —Datasets 2.4.0
  • —Tokenizers 0.12.1